def gather

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

import Base

21def gather(-A: Data, +values: List<&2, A>, axes: List<&2, Nat>) -> Maybe<&2, List<&2, A>>:
22  match axes:
23    case Nil{}:
24      Some{Nil{}}
25    case axis <> tail:
26      prepend(A, lookup(A, values, axis), gather(A, values, tail))