brainpy.losses.mean_squared_error
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).
keep_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.