def reorder

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

import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout

import Base

61def reorder(+values: List<&2, U32>, axes: List<&2, U32>) -> Maybe<&2, List<&2, U32>>:
62  match axes:
63    case Nil{}:
64      Some{Nil{}}
65    case axis <> tail:
66      reorder_head(value => reorder_tail(value, reorder(values, tail)), lookup(values, U32.to_nat(axis)))

Uses (7)

Used by (26)