Lutz Roeder 5 лет назад
Родитель
Сommit
31f22153dd
1 измененных файлов с 12 добавлено и 0 удалено
  1. 12 0
      source/pytorch-metadata.json

+ 12 - 0
source/pytorch-metadata.json

@@ -5232,5 +5232,17 @@
         { "name": "output", "type": "Tensor" }
       ]
     }
+  },
+  {
+    "name": "torch.einsum",
+    "schema": {
+      "inputs": [
+        { "name": "equation", "type": "string" },
+        { "name": "tensors", "type": "Tensor[]" }
+      ],
+      "outputs": [
+        { "name": "output", "type": "Tensor" }
+      ]
+    }
   }
 ]