def requested_axes

bendtorch-core@0.1.0.0 · src/tensor/transpose_spec.bend:41

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

41def requested_axes(+rank: Nat, +a: Nat, +b: Nat) -> List<&2, U32>:
42  Shape.words(natural_axes(rank, a, b))

Uses (5)

Used by (15)