def permute_source_index

bendtorch-core@0.1.0.0 · src/tensor/observation.bend:444

import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Observation

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

444# The row-major source index that `permute_source` names for output index
445# `index`, as a natural (0 where it names none).
446def permute_source_index(+input: Shape.Shape, +axes: List<&2, U32>, +index: Nat) -> Nat:
447  U32.to_nat(Maybe.default(&2, U32, permute_source(input, axes, U32.from_nat(index)), 0))

Uses (8)

Used by (21)