def map_index

bendtorch-core@0.1.0.0 · src/tensor/layout.bend:412

import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout

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

412def map_index(+input: Shape.Shape, +axes: List<&2, U32>, +plan: Plan<input, axes>, +i: U32) ->
413  Result<&2, &2, Error, Mapping<input, axes, plan, i>>:
414  map_decided(input, axes, plan, i, Shape.bool_case(U32.is_lt(i, Shape.count(output(input, axes, plan)))))

Uses (12)

Used by (24)