def contribution_offset

bendtorch-core@0.1.0.0 · src/autograd/reduce_fold.bend:18

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/contracts/broadcast.bend as Broadcast

18def contribution_offset(plan: Shape.BroadcastPlan, position: Nat) -> Nat:
19  Broadcast.right_map(plan, position)