![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
奶思图米球
这个作者很懒,什么都没留下…
展开
-
交叉熵损失(Cross-Entropy Loss)
在实际应用中,交叉熵损失通常与softmax函数结合使用,softmax函数可以将模型的原始输出(logits)转换为概率分布。对于正确的类别,当模型的预测概率越接近1时,损失越小;y:真实标签的概率分布,通常为一个独热编码向量,表示正确的类别位置为1,其余位置为0。对于错误的类别,当模型的预测概率越接近0时,损失越小;它鼓励模型对正确的类别有更高的预测概率,同时又对错误的类别有更低的预测概率。p:模型预测的概率分布,通常是模型输出层(如softmax层)的输出。Pi:预测概率向量中的第 i个元素。原创 2024-03-24 15:44:33 · 497 阅读 · 0 评论 -
torch.tensor()函数
在PyTorch中,张量是一个多维数组,它可以包含数值数据,并且可以在GPU或CPU上进行高效的数值计算。torch.tensor是创建张量的一种简单方法,创建一个torch.tensor时,你可以传递一系列的参数,包括数据、数据类型、设备(CPU或GPU)等,这个函数会根据提供的数据创建一个新的张量。PyTorch提供了多种创建张量的方式,包括torch.zeros、torch.ones、torch.randn等,这些函数可以创建具有特定数值或分布的张量。原创 2024-03-24 10:12:05 · 722 阅读 · 0 评论