def negated

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/reduction.bend as Reduction

222# -g, word by word: F32.neg flips the sign bit, as PyTorch's neg does.
223def negated(g: Core.Dynamic, zero: Result<&2, &1, Core.Error, Core.Tensor<Reduction.scalar_shape()>>) ->
224  Core.Dynamic & Result<&2, &1, Core.Error, Core.Dynamic>:
225  match zero:
226    case Fail{error}:
227      (g, Fail{error})
228    case Done{tensor}:
229      negated_pair(Elementwise.binary(Elementwise.NegateLeft{}, g, Core.pack(Reduction.scalar_shape(), tensor)))