ResizeLayer, small fix for dim tag

rwth-i6 · Dec 6, 2022 · f494763 · f494763
1 parent 4694a45
commit f494763
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/returnn/tf/layers/basic.py b/returnn/tf/layers/basic.py
@@ -7665,6 +7665,7 @@ def __init__(self, factor, axis, out_dim=None, kind="nn", fill_value=None, fill_
         else:
           raise TypeError("%s: unexpected factor_t %s" % (self, factor_t))
         self.output.dim_tags[axis].dyn_size_ext = dyn_size_ext
+        self.output.dim_tags[axis].set_tag_on_size_tensor(dyn_size_ext.placeholder, batch=dyn_size_ext.batch)
     self.output.placeholder = x
 
   def get_dep_layers(self):