diffusion_rs_common::nn::loss

Function mse

Source
pub fn mse(inp: &Tensor, target: &Tensor) -> Result<Tensor>
Expand description

The mean squared error loss.