LayerBase base out data, fixes for Data.sparse_dim

rwth-i6 · Nov 22, 2021 · aba4232 · aba4232
1 parent 208720b
commit aba4232
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/returnn/tf/layers/base.py b/returnn/tf/layers/base.py
@@ -324,7 +324,10 @@ def _base_get_out_data_from_opts(cls, network, name, out_type=None, n_out=NotSpe
       if sources_data:
         out_type.setdefault("batch_dim_axis", sources_data.batch_dim_axis)
         out_type.setdefault("time_dim_axis", sources_data.time_dim_axis)
-        if not out_type.get("sparse", False) and sources_data.feature_dim_axis_or_unspecified is not NotSpecified:
+        if (
+              not out_type.get("sparse", False) and
+              not out_type.get("sparse_dim", None) and
+              sources_data.feature_dim_axis_or_unspecified is not NotSpecified):
           if sources_data.feature_dim_axis_or_unspecified is not None:
             out_type.setdefault("feature_dim_axis", sources_data.feature_dim_axis_or_unspecified)
           else:  # None
@@ -334,7 +337,7 @@ def _base_get_out_data_from_opts(cls, network, name, out_type=None, n_out=NotSpe
         out_type.setdefault("time_dim_axis", None)
     if "shape" not in out_type and "dim_tags" not in out_type:
       if sources_data:
-        if out_type.get("sparse", False):
+        if out_type.get("sparse", False) or out_type.get("sparse_dim", None):
           out_type["dim_tags"] = sources_data.dim_tags_sparse
         else:  # not sparse
           feature_dim_axis = out_type.get("feature_dim_axis", NotSpecified)