Преглед на файлове

Update pytorch-metadata.json

Lutz Roeder преди 3 години
родител
ревизия
7225124e22
променени са 1 файла, в които са добавени 22 реда и са изтрити 2 реда
  1. 22 2
      source/pytorch-metadata.json

+ 22 - 2
source/pytorch-metadata.json

@@ -2339,7 +2339,7 @@
     ]
   },
   {
-    "name": "torch.round",
+    "name": "torch.round:",
     "inputs": [
       { "name": "self", "type": "Tensor" }
     ],
@@ -2348,7 +2348,7 @@
     ]
   },
   {
-    "name": "torch.round_",
+    "name": "torch.round_:",
     "inputs": [
       { "name": "self", "type": "Tensor" }
     ],
@@ -2356,6 +2356,26 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.round:decimals",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "decimals", "type": "int64" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.round_:decimals",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "decimals", "type": "int64" }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.select:int",
     "inputs": [