fit_a_nef.metrics.rmse

fit_a_nef.metrics.rmse(a: Array, b: Array) → Array

Vectorized version of rmse. Takes similar arguments as rmse but with additional array axes over which rmse is mapped.

Original documentation:

Returns the Mean Squared Error between a and b.

param a:

First image (or set of images).

type a:

jax.Array

param b:

Second image (or set of images).

type b:

jax.Array

return:

MSE between a and b.

rtype:

jax.Array