소스 검색

Update pytorch-metadata.json

Lutz Roeder 3 년 전
부모
커밋
e04b8ee30e
1개의 변경된 파일58개의 추가작업 그리고 1개의 파일을 삭제
  1. 58 1
      source/pytorch-metadata.json

+ 58 - 1
source/pytorch-metadata.json

@@ -2451,6 +2451,28 @@
       { "name": "indices", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.moveaxis:int",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "source", "type": "int64" },
+      { "name": "destination", "type": "int64" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.moveaxis:intlist",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "source", "type": "int64[]" },
+      { "name": "destination", "type": "int64[]" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.lstm:input",
     "category": "Layer",
@@ -2868,6 +2890,19 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.pad:",
+    "category": "Tensor",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "pad", "type": "int64[]" },
+      { "name": "mode", "type": "string", "default": "constant" },
+      { "name": "value", "type": "float32", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.permute",
     "category": "Shape",
@@ -5152,7 +5187,29 @@
     ]
   },
   {
-    "name": "torch.cumsum",
+    "name": "torch.cumsum:",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "int64" },
+      { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.cumsum:dimname",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "dim", "type": "Dimname" },
+      { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.cumsum_",
     "inputs": [
       { "name": "self", "type": "Tensor" },
       { "name": "dim", "type": "int64" },