def extent

bendtorch-core@0.1.0.0 · src/tensor/gather_LAWS.bend:28

import bendtorch-core@0.1.0.0/src/tensor/gather_LAWS.bend as GatherLAWS

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice

28def extent(+input: Shape.Shape, +at: Nat) -> Nat:
29  Slice.axis_extent(Shape.dimensions(input), at)

Uses (4)

Used by (4)