Explorar o código

Update pytorch-metadata.json

Lutz Roeder %!s(int64=3) %!d(string=hai) anos
pai
achega
d0c5b98db7
Modificáronse 1 ficheiros con 27 adicións e 0 borrados
  1. 27 0
      source/pytorch-metadata.json

+ 27 - 0
source/pytorch-metadata.json

@@ -6869,6 +6869,33 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.tril",
+    "category": "Layer",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "diagonal", "type": "int64", "default": 0 }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "torch.tril_indices",
+    "category": "Layer",
+    "inputs": [
+      { "name": "row", "type": "int64" },
+      { "name": "col", "type": "int64" },
+      { "name": "offset", "type": "int64", "default": 0 },
+      { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
+      { "name": "layout", "type": "Layout", "optional": true, "default": null },
+      { "name": "device", "type": "Device", "optional": true, "default": null },
+      { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.view_as_real",
     "category": "Layer",