要利用Torch構建自定義的損失函數,你需要定義一個新的損失函數類,并重寫其中的forward方法來計算損失值。接下來是一個示例代碼:
import torch
import torch.nn as nn
class CustomLoss(nn.Module):
def __init__(self):
super(CustomLoss, self).__init__()
def forward(self, pred, target):
loss = torch.mean((pred - target) ** 2) # 例如,這里定義為均方誤差損失函數
return loss
# 使用自定義的損失函數
criterion = CustomLoss()
pred = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5)
loss = criterion(pred, target)
print(loss)
在上面的示例中,我們首先定義了一個名為CustomLoss的新的損失函數類,然后在其中定義了forward方法來計算損失值。在這個例子中,我們將損失函數定義為均方誤差損失函數。最后,我們實例化了這個自定義的損失函數類,并計算了預測值和目標值之間的損失值。
通過這種方式,你可以通過Torch構建自定義的損失函數,并在模型訓練中使用它。