~/jendhub

def index_map

bendtorch-core@0.1.0.0 · src/shape/core.bend:857

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

857def index_map(+plan: BroadcastPlan) -> Index<extents(left_shape(plan)), extents(right_shape(plan)), extents(output_shape(plan))>:
858  +output = output_shape(plan)
859  index_of(left_shape(plan), right_shape(plan), output, bool_case(Nat.is_eq(elements(output), 0n)))

Uses (10)

Used by (38)