add

Author: metascroy

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

@@ -953,7 +953,7 @@ def gt(context, node):
     context.add(greater)
 
 
-@register_torch_op(torch_alias=["t", "numpy_t"])
+@register_torch_op(torch_alias=["t", "numpy_t", "transpose_copy"])
 def transpose(context, node):
     assert len(node.outputs) == 1
     inputs = _get_inputs(context, node)

coremltools/converters/mil/frontend/torch/test/test_torch_ops.py

@@ -7001,6 +7001,20 @@ def test(self, compute_unit, backend, frontend, shape, dims):
         )
 
 
+class TestTransposeCopy(TorchBaseTest):
+    @pytest.mark.parametrize(
+        "compute_unit, backend, frontend, shape, dims",
+        itertools.product(
+            compute_units, backends, frontends, COMMON_SHAPES, [(0, 1), (-2, -1), (1, 0), (-1, -2)]
+        ),
+    )
+    def test(self, compute_unit, backend, frontend, shape, dims):
+        model = ModuleWrapper(function=torch.transpose_copy, kwargs={"dim0": dims[0], "dim1": dims[1]})
+        self.run_compare_torch(
+            shape, model, compute_unit=compute_unit, backend=backend, frontend=frontend
+        )
+
+
 class TestTo(TorchBaseTest):
     @pytest.mark.parametrize(
         "compute_unit, backend, frontend",