def narrow_bwd

bendtorch@1.3.0.0 · tensor.bend:546

import Base

546# the gradient of narrow: dy in its window, zeros elsewhere
547def narrow_bwd(+s: List<&2, Nat>, +d: Nat, +st: Nat, +l: Nat, dy: Tensor<setd(s, d, l)>) -> TP<setd(s, d, l), s>:
548  match dy:
549    case Tensor{dd}:
550      tp_of(setd(s, d, l), s, place_l(numel(setd(s, d, l)), 0, narrow_map(s, d, st, l), Array.get(F32, dd, 0), new(numel(s))))