~/jendhub

def mean_axis

bendtorch@1.2.0.0 · tensor.bend:416

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

416# the mean over axis k: the sum divided by the axis' size
417def mean_axis(+s: List<&2, Nat>, +k: Nat, x: Tensor<s>) -> TP<s, remove(s, k)>:
418  tp_div(s, remove(s, k), n_f32(at1(s, k)), sum_axis(s, k, x))

Uses (9)

Used by (0)

No other definition uses this definition.