Skip to content

.forward()

Compute MSE loss.

Parameters:

  • p (Tensor) –

    Predictions tensor.

  • y (Tensor) –

    Targets tensor, same shape as p.