Skip to content

Commit 208720b

Browse files
committed
CombineLayer, fix for Data.sparse_dim
1 parent 0fc68e5 commit 208720b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

returnn/tf/layers/basic.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6295,6 +6295,11 @@ def get_out_data_from_opts(cls, eval_locals=None, n_out=NotSpecified, out_type=N
62956295
out_type_.pop("batch_dim_axis", None)
62966296
out_type_.pop("feature_dim_axis", None)
62976297
out_type_.pop("time_dim_axis", None)
6298+
if "dim" in out_type:
6299+
if out_type_.pop("sparse_dim", None):
6300+
out_type_["sparse"] = True
6301+
if "sparse" in out_type:
6302+
out_type_.pop("sparse_dim", None)
62986303
out_type_.update(out_type)
62996304
elif callable(out_type):
63006305
def call_out_type_with_eval_locals(**out_type_kwargs):

0 commit comments

Comments
 (0)