law permutation_count

bendtorch-core@0.1.0.0 · src/tensor/transform_LAWS.bend:24

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

24law permutation_count:
25  for +input: Shape.Shape
26  for +axes: List<&2, U32>
27  for +plan: Layout.Plan<input, axes>
28  {Shape.logical_numel(Shape.dimensions(input)) == Shape.logical_numel(Shape.dimensions(Layout.output(input, axes, plan))) : Nat}