|
|
@@ -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" },
|