def permutation_checked

bendtorch-core@0.1.0.0 · src/tensor/permutation_mapping.bend:14

import bendtorch-core@0.1.0.0/src/tensor/permutation_mapping.bend as PermutationMapping
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout

14def permutation_checked(+input: Shape.Shape, +axes: List<&2, U32>, plan: Layout.Plan<input, axes>) ->
15  {True{} == Layout.bijective(input, axes) : Bool}:
16  Layout.Permutation{out, valid, ordered, count} = plan
17  valid

Uses (6)

Used by (3)