def ranked

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

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

3017def ranked(-A: Data, -rank: Nat, -axes: List<&2, Nat>, -items: List<&2, A>, certificate: ListFacts<A, rank, axes, items>) ->
3018  {List.length(&2, A, Axis.order(A, items, axes)) == rank : Nat}:
3019  match certificate:
3020    case Lists{selected, ranked, scattered, restored, reverse_restored, scatter_restored}:
3021      ranked