当输出是浮点数数组时,pytorch中适当的损失函数

问题描述

我正在编写与https://pytorch.org/tutorials/intermediate/seq2seq_translation_tutorial.html非常相似的编码器/解码器模型

唯一的区别是,此处的单词由某些索引表示。我想根据另一个指标来显示它们,该指标由拍号表示。

loss function nn.criterion = nn.NLLLoss(),似乎在我们只在使用类时才起作用。

如果我的输出数组不是整数数组,而是浮点数数组,那么我可以使用哪种损失函数?考虑到所有其他部分都与本教程相似?

谢谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)