def gather_ranked

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:173

import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

import Base

173def gather_ranked(+dimensions: List<&2, U32>, +axis: Nat, +index: List<&2, U32>, +entries: List<&2, U32>, same: Bool) ->
174  Result<&2, &2, Error, Planned>:
175  match same:
176    case False{}:
177      Fail{IndexRankMismatch{}}
178    case True{}:
179      gather_shaped(dimensions, axis, index, entries, matches(dimensions, index, axis))

Uses (9)

Used by (5)