Skip to content

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.