def permute

import bendtorch@1.3.0.0/tensor.bend as Tensor

import Base

507def permute(+s: List<&2, Nat>, +p: List<&2, Nat>, -pf: {True{} == is_perm(p, rank(s)) : Bool}, x: Tensor<s>) -> TP<s, permuted(s, p)>:
508  match x:
509    case Tensor{d}:
510      tp_of(s, permuted(s, p), gather_l(numel(s), 0, perm_map(s, p), Array.get(F32, d, imap(perm_map(s, p), 0)), new(numel(s))))

Uses (16)

Bool
List
Nat
F32
Array.get
…

Used by (0)

No other definition uses this definition.