pytorch 如何自定义损失函数

我是大佬们的搬运工

在做神经网络的时候，一般情况我们可以直接调用pytorch提供的函数损失函数，例如：

class torch.nn.MSELoss(size_average=True)

调用方式：

creterion=torch.nn.MSELoss()

loss=creterion(x,y)

 

想要自定义损失函数，就要按照损失函数的格式来编写

观察自带的损失函数，我们要写的损失函数应该也是这种样子的

又参考了知乎上大佬的分享

虽说大佬给了三种方法，但是我觉得像我这种平头小老百姓其实用得到的只有第一种：