def axis_limit

bendtorch-core@0.1.0.0 · src/tensor/transpose_spec.bend:5

import Base

5def axis_limit(rank: Nat) -> Nat:
6  match rank:
7    case 0n:
8      1n
9    case 1n+p:
10      1n+p