Skip to content

Commit 0fc68e5

Browse files
committed
GatherNdLayer, fix for Data.sparse_dim
1 parent ee6e03d commit 0fc68e5

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

returnn/tf/layers/basic.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1400,8 +1400,7 @@ def get_out_data_from_opts(cls, name, sources, position, **kwargs):
14001400
if position_data.time_dim_axis is None:
14011401
if input_data.time_dim_axis is not None and input_data.time_dim_axis_excluding_batch >= 1:
14021402
out_type["time_dim_axis"] = len(dim_tags) + input_data.time_dim_axis_excluding_batch - 2
1403-
out_type["dim"] = input_data.dim
1404-
out_type["sparse"] = input_data.sparse
1403+
out_type["sparse_dim"] = input_data.sparse_dim
14051404
out_type["dtype"] = input_data.dtype
14061405
out_type["beam"] = SearchBeam.get_combined_beam(position_data.beam, input_data.beam)
14071406
return Data(**out_type)

0 commit comments

Comments
 (0)