law transpose_axes_self_inverse

bendtorch-core@0.1.0.0 · src/tensor/transpose_LAWS.bend:23

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/transpose_spec.bend as Transpose

23law transpose_axes_self_inverse:
24  for +shape: Shape.Shape
25  for +a: U32
26  for +b: U32
27  for +inside: {True{} == Transpose.valid(shape, a, b) : Bool}
28  {Axis.inverse_axes(Shape.rank_dimensions(Shape.dimensions(shape)), Shape.naturals(Layout.transpose_axes(Shape.dimensions(shape), a, b, 0))) ==
29    Shape.naturals(Layout.transpose_axes(Shape.dimensions(shape), a, b, 0)) : List<&2, Nat>}