def select

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

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

import Base

49def select(-A: Data, +values: List<&2, A>, index: Maybe<&2, Nat>) -> Maybe<&2, A>:
50  match index:
51    case None{}:
52      None{}
53    case Some{value}:
54      lookup(A, values, value)

Uses (4)

Used by (8)