TF test_reclayer_optimize_out_pos_enc_layer

rwth-i6 · Sep 13, 2023 · 816492a · 816492a
1 parent 3784a39
commit 816492a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_TFNetworkRecLayer.py b/tests/test_TFNetworkRecLayer.py
@@ -7304,6 +7304,19 @@ def test_reclayer_optimize_out_cumsum_unrelated_axis():
     )
 
 
+def test_reclayer_optimize_out_pos_enc_layer():
+    feat_dim = FeatureDim("feat", dimension=11)
+    check_reclayer_optimize_out(
+        feat_dim=feat_dim,
+        subnet_layer_dict={
+            "class": "positional_encoding",
+            "out_dim": feat_dim,
+            "from": "data:source",
+            "add_to_input": True,
+        },
+    )
+
+
 def test_reclayer_optimize_out_rel_pos_enc_layer():
     # https://github.com/rwth-i6/returnn/issues/1253
     time_dim = SpatialDim("time")