def mse_loss

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

import Base

1637# Records mse_loss(a, b), PyTorch's mse_loss(input, target) with mean
1638# reduction, a rank-0 tensor, from copies of the operands' values. The
1639# differences a - b and b - a, over the shapes' broadcast, are saved for the
1640# VJP. Shapes that do not broadcast are refused with IncompatibleBroadcast
1641# (6); an unknown id with UnknownNode; the tape comes back unchanged.
1642def mse_loss(tape: Tape, +a: Nat, +b: Nat) -> Tape & Result<&2, &2, Error, Nat>:
1643  binary(tape, SquaredError{}, a, b)