Torch.nn.MSELoss

torch.nn.MSELoss(size_average=None, reduce=None, reduction='mean')

import torch.nn as nn

criterion = nn.MSELoss()
...
loss = criterion(input, target)