python 和 torch 交叉熵损失

import math
import numpy as np   # 导入 math 模块
import torch

x=0.9
y=0.1
print("math.log",-x* math.log(1-y))
print("n  p.log",-x* np.log(1-y))


x_tensor = torch.tensor([x])
y_tensor = torch.tensor([y])


loss_tensor = (- x_tensor * torch.log(1-y_tensor)).sum()
print(loss_tensor)
发布了2608 篇原创文章 · 获赞 920 · 访问量 506万+

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/103811407