test_CombineLayer_match_unknown_derived

Author: albertz

tests/test_TFNetworkLayer.py

@@ -718,6 +718,24 @@ def test_CombineLayer_match_unknown():
     assert out.dim_tags[:2] == dat2.dim_tags[:2] and out.batch_shape == dat2.batch_shape
 
 
+def test_CombineLayer_match_unknown_derived():
+  with make_scope() as session:
+    dat1 = Data(name="undefined", shape=(None, 3))
+    assert dat1.dim_tags[1].undefined
+    dat1_derived_dim_tags = list(dat1.dim_tags)
+    dat1_derived_dim_tags[1] = DimensionTag(
+      kind=DimensionTag.Types.Spatial, description="undefined_derived_dim", derived_from_tag=dat1.dim_tags[1])
+    dat1_derived = Data(name="undefined_derived", dim_tags=dat1_derived_dim_tags)
+    assert dat1_derived.dim_tags[1].undefined
+    # Create placeholders to have this dyn size clearly defined.
+    dat2 = Data(name="defined", shape=(None, 3), auto_create_placeholders=True)
+    net = TFNetwork(extern_data=ExternData())
+    layer1 = InternalLayer(name="layer1_undefined_derived", network=net, output=dat1_derived)
+    layer2 = InternalLayer(name="layer2_defined", network=net, output=dat2)
+    out = CombineLayer.get_out_data_from_opts(name="combine", network=net, sources=[layer1, layer2])
+    assert out.dim_tags[:2] == dat2.dim_tags[:2] and out.batch_shape == dat2.batch_shape
+
+
 def test_CombineLayer_different_batch_axis():
   # ["base:enc_ctx", "weight_feedback", "s_transformed"]
   # base:enc_ctx: Data(name='enc_ctx_output', shape=(None, 14), batch_dim_axis=1)