meddlr.metrics.functional.rmse#

meddlr.metrics.functional.rmse(pred: Tensor, target: Tensor, im_type: Optional[str] = None) → Tensor[source]#

Computes root mean square error.

Parameters

  • pred (torch.Tensor) – The prediction. Either a complex or real tensor.

  • target (torch.Tensor) – The target. Either a complex or real tensor.

  • im_type (str, optional) – The image type to compute metric on. This only applies for complex inputs, otherwise ignored. Either 'magnitude' (default) to compute metric on magnitude images or 'phase' to compute metric on phase/angle images. If None, computed on complex images.

Returns

The root mean square error.

Return type

torch.Tensor