def swapped_axes

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

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

44# The permutation `transpose_copy(input, a, b)` runs: the identity of the
45# input's rank with axes a and b swapped.
46def swapped_axes(+input: Shape.Shape, +a: U32, +b: U32) -> List<&2, U32>:
47  requested_axes(Shape.rank_dimensions(Shape.dimensions(input)), U32.to_nat(a), U32.to_nat(b))