~/jendhub

def reordered_elements_same

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

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
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis
import bendtorch-core@0.1.0.0/src/tensor/axis_algebra.bend as AxisAlgebra

198def reordered_elements_same(+input: Shape.Shape, +axes: List<&2, U32>, +output: Shape.Shape,
199  +valid: {True{} == Layout.bijective(input, axes) : Bool},
200  preserved: {Shape.dimensions(output) == Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes)) : List<&2, U32>}) ->
201  {Shape.elements(input) == Shape.elements(output) : Nat}:
202  element_counts_same(input, output,
203    Equal.trans(Nat, Shape.logical_numel(Shape.dimensions(input)),
204      Shape.logical_numel(Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes))), Shape.logical_numel(Shape.dimensions(output)),
205      Equal.sym(Nat, Shape.logical_numel(Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes))), Shape.logical_numel(Shape.dimensions(input)),
206        AxisAlgebra.permutation_product(input, axes, valid)),
207      Equal.cong(List<&2, U32>, Nat, Shape.logical_numel, Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes)), Shape.dimensions(output),
208        Equal.sym(List<&2, U32>, Shape.dimensions(output), Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes)), preserved))))

Uses (16)

Used by (1)