def right_map

bendtorch-core@0.1.0.0 · src/tensor/contracts/broadcast.bend:30

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

30def right_map(plan: Shape.BroadcastPlan, index: Nat) -> Nat:
31  +plan = plan
32  right_walk(Shape.extents(Shape.left_shape(plan)), Shape.extents(Shape.right_shape(plan)), Shape.extents(Shape.output_shape(plan)),
33    Shape.index_map(plan), index)

Uses (8)

Used by (15)