def extent
import bendtorch-core@0.1.0.0/src/tensor/signed.bend as Signedimport Base
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice
306# The extent of axis `axis`, 0 past the rank.
307def extent(dimensions: List<&2, U32>, +axis: U32) -> U32:
308 U32.from_nat(Slice.extent_at(dimensions, axis))