Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
aresnow1 committed Dec 19, 2022
1 parent 2fde257 commit 5b3f738
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mars/tensor/base/transpose.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
def _reorder(x, axes):
if x is None:
return
return type(x)(np.array(x)[list(axes)].tolist())
return type(x)(x[ax] for ax in axes)


class TensorTranspose(TensorHasInput, TensorOperandMixin):
Expand Down

0 comments on commit 5b3f738

Please sign in to comment.