mean_squared_error

class brainpy.losses.mean_squared_error(predicts, targets, axis=None, reduction='mean')

Computes the mean squared error between x and y.

Parameters:

• predicts – a tensor of shape (d0, .. dN-1).

• targets – a tensor of shape (d0, .. dN-1).

• axis – a sequence of the dimensions to keep, use None to return a scalar value.

Return type:

tensor of shape (d_i, ..., for i in keep_axis) containing the mean squared error.