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

+ 31 - 0
source/pytorch-metadata.json

@@ -5550,6 +5550,37 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.linalg_solve:",
+    "attributes": [
+      { "name": "input", "type": "Tensor" },
+      { "name": "other", "type": "Tensor" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.linalg_tensorinv:",
+    "attributes": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "ind", "type": "int64", "default": 2 }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.linalg_tensorsolve:",
+    "attributes": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "other", "type": "Tensor" },
+      { "name": "dims", "type": "int64[]", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.linspace",
     "attributes": [