def gather_found

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1581

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

1581def gather_found(+a: Nat, +indexing: Indexing, pair: Found()) -> Tape & Result<&2, &2, Error, Nat>:
1582  (entries, input) = pair
1583  match input:
1584    case Fail{error}:
1585      (Tape{entries}, Fail{UnknownNode{}})
1586    case Done{input}:
1587      gather_shaped(a, indexing, entries, Core.shape(input))