Tensor.Sum¶
Name | Mandatory | Description | Default | Type |
---|---|---|---|---|
⬅️ Input |
The input of the shard, if any | Tensor |
||
Output ➡️ |
The resulting output of the shard | Tensor |
||
Dims |
No | The dimensions to sum over. If not provided, sum over all dimensions. | none |
[Int] Var([Int]) None |
Computes the sum of tensor elements along the specified dimensions.