def lookup

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

import Base

11def lookup(-A: Data, values: List<&2, A>, index: Nat) -> Maybe<&2, A>:
12  first(A, List.drop(&2, A, values, index))

Uses (5)