def capacity_of
import bendtorch-core@0.1.0.0/src/autograd/reduce_PROOF.bend as ReducePROOFimport Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
166def capacity_of(+shape: Shape.Shape) -> Shape.LE(Shape.logical_numel(Shape.dimensions(shape)), 1048576n):
167 match shape:
168 case Shape.SShape{dims, size, exact, rank_bound, capacity_bound}:
169 capacity_bound