def extent_at
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Sliceimport Base
100def extent_at(dimensions: List<&2, U32>, +dim: U32) -> Nat:
101 axis_extent(dimensions, U32.to_nat(dim))
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Sliceimport Base
100def extent_at(dimensions: List<&2, U32>, +dim: U32) -> Nat:
101 axis_extent(dimensions, U32.to_nat(dim))