def capacity_of
import bendtorch-core@0.1.0.0/src/autograd/broadcast_PROOF.bend as BroadcastPROOFimport Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
292def capacity_of(+shape: Shape.Shape) -> Shape.LE(Shape.logical_numel(Shape.dimensions(shape)), 1048576n):
293 match shape:
294 case Shape.SShape{dims, size, exact, rank_bound, capacity_bound}:
295 capacity_bound