SliceNdLayer added set_tag_on_size_tensor for dynamic size case

rwth-i6 · Sep 17, 2021 · 85b48b0 · 85b48b0
1 parent 5eb93b5
commit 85b48b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/returnn/tf/layers/basic.py b/returnn/tf/layers/basic.py
@@ -901,6 +901,7 @@ def __init__(self, start, size, min_size=None, **kwargs):
         beam=slice_tag.batch.beam if slice_tag.batch else self.output.beam,
         control_flow_ctx=slice_tag.control_flow_ctx)
       slice_tag.dyn_size_ext = dyn_size_ext
+      slice_tag.set_tag_on_size_tensor(dyn_size)
     gather_positions_data = start_data.copy_template(name="%s_gather_positions" % self.name)
     gather_positions_data = gather_positions_data.copy_add_dim_by_tag(
       slice_tag,