add torch view as reshape operator (#1368)

sony · Mar 4, 2025 · a6a4dd8 · a6a4dd8
1 parent 51be057
commit a6a4dd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...mpression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py b/...mpression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py
@@ -70,7 +70,7 @@ def __init__(self):
             OperatorSetNames.HARDSWISH: [Hardswish, hardswish],
             OperatorSetNames.FLATTEN: [Flatten, flatten],
             OperatorSetNames.GET_ITEM: [operator.getitem],
-            OperatorSetNames.RESHAPE: [reshape],
+            OperatorSetNames.RESHAPE: [reshape, torch.Tensor.view],
             OperatorSetNames.UNSQUEEZE: [unsqueeze],
             OperatorSetNames.SQUEEZE: [squeeze],
             OperatorSetNames.PERMUTE: [permute],