brainpy.losses.mean_squared_error#

brainpy.losses.mean_squared_error(predicts, targets, axis=None)[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.