损失函数的定义为:
loss = nn.MarginRankingLoss(margin=0.5,reduction='none')
#reduction :[ 'none' | 'mean' | 'sum']
损失函数的使用:
x1 = torch.tensor([1,2,3,4],requires_grad=True, dtype=torch.float32)
x2 = torch.tensor([3,4,5,6
损失函数的定义为:
loss = nn.MarginRankingLoss(margin=0.5,reduction='none')
#reduction :[ 'none' | 'mean' | 'sum']
损失函数的使用:
x1 = torch.tensor([1,2,3,4],requires_grad=True, dtype=torch.float32)
x2 = torch.tensor([3,4,5,6