Jelajahi Sumber

Update pytorch-metadata.json

Lutz Roeder 3 tahun lalu
induk
melakukan
a65931dc21
1 mengubah file dengan 14 tambahan dan 0 penghapusan
  1. 14 0
      source/pytorch-metadata.json

+ 14 - 0
source/pytorch-metadata.json

@@ -6909,6 +6909,20 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.cross_entropy_loss",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "target", "type": "Tensor" },
+      { "name": "weight", "type": "Tensor", "optional": true, "default": null },
+      { "name": "reduction", "type": "int64" },
+      { "name": "ignore_index", "type": "int64", "default": -100 },
+      { "name": "label_smoothing", "type": "float32", "default": 0 }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.binary_cross_entropy_with_logits",
     "inputs": [