def contains

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

import Base

86def contains(+axis: Nat, axes: List<&2, Nat>) -> Bool:
87  match axes:
88    case Nil{}:
89      False{}
90    case head <> tail:
91      Bool.or(Nat.is_eq(axis, head), contains(axis, tail))