def contains

bendtorch-core@0.1.0.0 · src/tensor/layout.bend:13

import Base

13def contains(+axis: U32, axes: List<&2, U32>) -> Bool:
14  match axes:
15    case Nil{}:
16      False{}
17    case +head <> tail:
18      Bool.or(U32.is_eq(axis, head), contains(axis, tail))