Преглед изворни кода

Update pytorch-metadata.json

Lutz Roeder пре 5 година
родитељ
комит
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" }
+      ]
+    }
   }
 ]