def selected

bendtorch-core@0.1.0.0 · src/tensor/axis_algebra.bend:3011

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

3011def selected(-A: Data, -rank: Nat, -axes: List<&2, Nat>, -items: List<&2, A>, certificate: ListFacts<A, rank, axes, items>) ->
3012  {Axis.gather(A, items, axes) == Some{Axis.order(A, items, axes)} : Maybe<&2, List<&2, A>>}:
3013  match certificate:
3014    case Lists{selected, ranked, scattered, restored, reverse_restored, scatter_restored}:
3015      selected