~/jendhub

def mean_axis_bwd

bendtorch@1.3.0.0 · tensor.bend:422

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

422# the gradient of mean_axis: dy divided by the axis' size, copied along it
423def mean_axis_bwd(+s: List<&2, Nat>, +k: Nat, dy: Tensor<remove(s, k)>) -> TP<remove(s, k), s>:
424  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.