pytorch中的top

2024-06-13 15:43| 来源: 网络整理| 查看: 265

PyTorch损失函数之交叉熵损失函数nn.CrossEntropyLoss()

Z_0_0: 感谢，我把当时给input加的softmax去掉之后就好使了，这个问题困扰我一天了

PyTorch损失函数之交叉熵损失函数nn.CrossEntropyLoss()

SWANSSSED: 总体写得不错，不过这里有点需要纠正一下，nn.CrossEntropyLoss函数可以接受one-hot形式的target编码输入，也可以接受标签单个数字形式的输入。

PyTorch损失函数之交叉熵损失函数nn.CrossEntropyLoss()

Starry0716: 是的，第0维仍是之前的0维，没有变

torch.Tensor.index_add_函数，pytorch中的tf.unsorted_segment_sum

苦逼学习pytorch中: index的值必须介于0和C-1之间,这个应该是介于0到N-1之间把

PyTorch损失函数之交叉熵损失函数nn.CrossEntropyLoss()

myaijarvis: 此处的意思是（第0维的数据换成第1维的），第1维换成第2维，。。。括号里面应该是不要的注释

【本文地址】

公司简介

联系我们