law scalar_mse_grad

bendtorch-core@0.1.0.0 · src/tensor/loss_ops_LAWS.bend:57

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

57# PyTorch's mse_loss backward: (scale * d) * g, scale = 2 / N, for g on the
58# left and d on the right.
59law scalar_mse_grad:
60  for +scale: F32
61  for +left: F32
62  for +right: F32
63  {Elementwise.scalar(Elementwise.MseGrad{scale}, left, right) == F32.mul(F32.mul(scale, right), left) : F32}