~/jendhub

def sum_all_bwd

bendtorch@1.3.0.0 · tensor.bend:436

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

436def sum_all_bwd(+s: List<&2, Nat>, dy: Tensor<[]>) -> TP<[], s>:
437  match dy:
438    case Tensor{d}:
439      tp_of([], s, gather_l(numel(s), 0, all_map(s), Array.get(F32, d, 0), new(numel(s))))

Uses (11)

Used by (1)