def scatter_sent

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

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
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice

633def scatter_sent(+a: Nat, +input: Shape.Shape, pair: Core.Dynamic & Result<&2, &1, Slice.Error, Core.Dynamic>) -> Pass:
634  (g, result) = pair
635  match result:
636    case Fail{error}:
637      Pass{Unsaved{}, g, Fail{SliceError{error}}}
638    case Done{scattered}:
639      Pass{Unsaved{}, g, Done{[Contribution{a, input, scattered}]}}