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

+ 76 - 0
source/pytorch-metadata.json

@@ -4782,6 +4782,18 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.scatter:src",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Tensor" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.scatter_:src",
     "inputs": [
@@ -4794,6 +4806,18 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.scatter:value",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Scalar" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.scatter_:value",
     "inputs": [
@@ -4806,6 +4830,58 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.scatter:reduce",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Tensor" },
+      { "name": "reduce", "type": "string" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.scatter_:reduce",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Tensor" },
+      { "name": "reduce", "type": "string" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.scatter:value_reduce",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Scalar" },
+      { "name": "reduce", "type": "string" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.scatter_:value_reduce",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "index", "type": "Tensor" },
+      { "name": "src", "type": "Scalar" },
+      { "name": "reduce", "type": "string" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.scatter_add_",
     "inputs": [