~/jendhub

def gather_shaped

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

import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

1576def gather_shaped(+a: Nat, +indexing: Indexing, entries: List<&1, Entry>, pair: Core.Dynamic & Shape.Shape) ->
1577  Tape & Result<&2, &2, Error, Nat>:
1578  (input, +shape) = pair
1579  gather_taken(a, shape, indexing, entries, indexing_value(input, indexing))

Uses (13)

Used by (1)