def natural_axes

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

import bendtorch-core@0.1.0.0/src/tensor/transpose_spec.bend as TransposeSpec
import Base
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis

38def natural_axes(+rank: Nat, +a: Nat, +b: Nat) -> List<&2, Nat>:
39  swap(Nat, Axis.indices(rank, 0n), a, b)

Uses (4)

Used by (1)