def position

bendtorch-core@0.1.0.0 · src/tensor/axis_spec.bend:42

import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as AxisSpec

import Base

42def position(axes: List<&2, Nat>, +axis: Nat) -> Maybe<&2, Nat>:
43  match axes:
44    case Nil{}:
45      None{}
46    case head <> tail:
47      position_decided(Nat.is_eq(head, axis), position(tail, axis))

Uses (5)

Used by (5)