코딩걸음마

import torch def mse(x_hat, x): # |x_hat| = (batch_size, dim) # |x| = (batch_size, dim) y = ((x - x_hat)**2).mean() return y x = torch.FloatTensor([[1, 1], [2, 2]]) x_hat = torch.FloatTensor([[0, 0], [0, 0]]) print(x.size(), x_hat.size()) torch.Size([2, 2]) torch.Size([2, 2]) mse(x_hat, x) tensor(2.5000) MSE in PyTorch import torch.nn.functional as F F.mse_loss(x_hat, x) tensor(2.5000) reduction..