def axis_limit

bendtorch-core@0.1.0.0 · src/tensor/signed.bend:77

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/transpose_spec.bend as Transpose

77# The axes of the input for transpose and sum: its rank, and 1 for a rank-0
78# input, whose only axis is 0 (or -1).
79def axis_limit(+input: Shape.Shape) -> U32:
80  U32.from_nat(Transpose.axis_limit(Shape.rank_dimensions(Shape.dimensions(input))))