def transpose_axes

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

import Base

167def transpose_axes(dimensions: List<&2, U32>, +a: U32, +b: U32, +i: U32) -> List<&2, U32>:
168  match dimensions:
169    case Nil{}:
170      Nil{}
171    case dimension <> tail:
172      Bool.pick(U32, U32.is_eq(i, a), b, Bool.pick(U32, U32.is_eq(i, b), a, i)) <>
173        transpose_axes(tail, a, b, U32.inc(i))