~/jendhub

def sum_all

bendtorch@1.3.0.0 · tensor.bend:430

import bendtorch@1.3.0.0/tensor.bend as Tensor
import Base

430# the sum of every element: a tensor of shape [] (one number)
431def sum_all(+s: List<&2, Nat>, x: Tensor<s>) -> TP<s, []>:
432  match x:
433    case Tensor{d}:
434      tp_of(s, [], scatter_l(numel(s), 0, all_map(s), Array.get(F32, d, 0), Array.get(F32, new(1n), 0)))

Uses (11)

Used by (1)