def extent_of
bendtorch-core@0.1.0.0 · src/tensor/signed_LAWS.bend:320
import bendtorch-core@0.1.0.0/src/tensor/signed_LAWS.bend as SignedLAWS
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice
320def extent_of(input: Core.Dynamic, +axis: U32) -> U32:
321 U32.from_nat(Slice.extent_at(Shape.dimensions(Totality.dynamic_shape(input)), axis))