~/jendhub

def mean_axis_bwd

bendtorch@1.2.0.0 · tensor.bend:420

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

420# the gradient of mean_axis: dy divided by the axis' size, copied along it
421def mean_axis_bwd(+s: List<&2, Nat>, +k: Nat, dy: Tensor<remove(s, k)>) -> TP<remove(s, k), s>:
422  tp_div(remove(s, k), s, n_f32(at1(s, k)), sum_axis_bwd(s, k, dy))

Uses (9)

Used by (0)

No other definition uses this definition.