def swap

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

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

34def swap(-A: Data, +items: List<&2, A>, +a: Nat, +b: Nat) -> List<&2, A>:
35  left right = Axis.lookup(A, items, a) Axis.lookup(A, items, b)
36  swap_selected(A, items, a, b, left, right)