Torch中怎么定义一个损失函数

在Torch中定义一个损失函数，一般是通过继承nn.Module类来实现的。以下是一个示例：import torchimport torch.nn as nnclass CustomLoss(nn.Module):def __init__(self):super(CustomLoss, self).__init__()def forward(self, output, target):loss