Skip to content

Commit 6d84d64

Browse files
Update dpctl/tensor/_manipulation_functions.py
1 parent 7b51457 commit 6d84d64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpctl/tensor/_manipulation_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,6 @@ def expand_dims(X, axes):
8181
axes = normalize_axis_tuple(axes, out_ndim)
8282

8383
shape_it = iter(X.shape)
84-
shape = [1 if ax in axes else next(shape_it) for ax in range(out_ndim)]
84+
shape = tuple(1 if ax in axes else next(shape_it) for ax in range(out_ndim))
8585

8686
return dpt.reshape(X, shape)

0 commit comments

Comments
 (0)