def expanded

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

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/autograd/broadcast.bend as Broadcast

268def expanded(+a: Nat, +input: Shape.Shape, pair: Core.Dynamic & Result<&2, &1, Broadcast.Error, Core.Dynamic>) -> Pass:
269  (g, result) = pair
270  match result:
271    case Fail{error}:
272      Pass{Unsaved{}, g, Fail{BroadcastError{error}}}
273    case Done{expanded}:
274      Pass{Unsaved{}, g, Done{[Contribution{a, input, expanded}]}}