ソースを参照

Update pytorch-metadata.json

Lutz Roeder 4 年 前
コミット
fe501ce3b5
1 ファイル変更11 行追加0 行削除
  1. 11 0
      source/pytorch-metadata.json

+ 11 - 0
source/pytorch-metadata.json

@@ -2331,6 +2331,17 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.rot90",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "k", "type": "int64", "default": 1 },
+      { "name": "dims", "type": "int64[]", "default": [ 0, 1 ] }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.round",
     "inputs": [