woomay的博客

pytorch MSELoss参数详解import torchimport numpy as nploss_fn = torch.nn.MSELoss(reduce=False, size_average=False)a=np.array([[1,2],[3,8]])b=np.array([[5,4],[6,2]])input = torch.autograd.Variable(to...

pytorch CrossEntropyLossimport torchimport torch.nn as nnimport mathloss = nn.CrossEntropyLoss()input = torch.randn(3, 5, requires_grad=True)print('输入：')print(input,input.shape)target = torch...