Data copy_compatible_to TODO replace dim tag

rwth-i6 · Mar 4, 2022 · 6723531 · 6723531
1 parent fcecd20
commit 6723531
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/returnn/tf/util/data.py b/returnn/tf/util/data.py
@@ -3843,6 +3843,7 @@ def copy_replace_dim_tag(self, axis, new_dim_tag):
         mask_t = mask.placeholder  # [T,B], e.g. [1,0,1] (ignoring batch-dim for example)
         # Also see UnmaskLayer.
         idxs = tf.cumsum(tf.cast(mask_t, tf.int32), axis=0)  # [T,B], e.g. [1,1,2]
+        # TODO need to get initial ... how?
         initial_bc = tf.reshape(0., shape=[1] * self.batch_ndim)  # add bc dims
         src = self
         if axis != 0: