KLDivLoss实例

下面是一个使用nn.KLDivLoss()的示例：

import torch
import torch.nn as nn

# 创建KLDivLoss对象
criterion = nn.KLDivLoss()

# 准备输入数据
input = torch.tensor([[0.2, 0.3, 0.5]])  # 输入数据
target = torch.tensor([[0.1, 0.5, 0.4]])  # 目标数据

# 计算KLDivLoss
loss = criterion(torch.log(input), target)
print(loss)

输出:

tensor(0.1155)

在这个例子中，我们使用了nn.KLDivLoss()来计算input和target之间的KL散度损失。首先，我们需要对输入数据input和目标数据target进行对数变换，然后将它们作为参数传递给criterion函数。最后，我们得到了计算得到的KL散度损失值loss。