import torch
import numpy as np
regression_loss = torch.nn.MSELoss(reduce=False, size_average=False)
inputs = torch.autograd.Variable(torch.randn(3, 5))
target = torch.autograd.Variable(torch.randn(3, 5))
loss = regression_loss(inputs, target)
print(inputs)
print('\n')
print(target)
print('\n')
print(loss)
print('\n')
print(inputs.size(), target.size(), loss.size())