mean_squared_error

mean_squared_error#

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

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.

Returns:

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