fit_a_nef.metrics.mse

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

Vectorized version of mse. Takes similar arguments as mse but with additional array axes over which mse 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