law flatten_unflatten

bendtorch-core@0.1.0.0 · src/shape/LAWS.bend:129

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

129law flatten_unflatten:
130  for +dims: List<&2, Nat>
131  for +index: Shape.Fin(Shape.numel(dims))
132  {Shape.flatten(dims, Shape.unflatten(dims, index)) == index : Shape.Fin(Shape.numel(dims))}