Skip to content

Commit 7a07062

Browse files
authored
Add numpy_t as alias of transpose (#2046)
1 parent e72db64 commit 7a07062

File tree

1 file changed

+1
-1
lines changed
  • coremltools/converters/mil/frontend/torch

1 file changed

+1
-1
lines changed

coremltools/converters/mil/frontend/torch/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -788,7 +788,7 @@ def gt(context, node):
788788
context.add(greater)
789789

790790

791-
@register_torch_op(torch_alias=["t"])
791+
@register_torch_op(torch_alias=["t", "numpy_t"])
792792
def transpose(context, node):
793793
assert len(node.outputs) == 1
794794
inputs = _get_inputs(context, node)

0 commit comments

Comments
 (0)