【问题标题】:What function is the equivalent of torch.nn.CrossEntropyLoss in TF2.0 for multiclass logits对于多类 logits,什么函数相当于 TF2.0 中的 torch.nn.CrossEntropyLoss
【发布时间】:2021-09-16 22:48:39
【问题描述】:

对不起,我问这个问题,它看起来很愚蠢,但我没有其他方法可以寻找它。如果您有一些可以轻松比较此类问题的并行文档,请告诉我。

【问题讨论】:

    标签: pytorch tensorflow2.0 torch


    【解决方案1】:

    SO的好人,SparseCategoricalCrossentropy相当于TF2.0中的它:https://www.tensorflow.org/api_docs/python/tf/keras/losses/SparseCategoricalCrossentropy

    它是 SparseCategoricalCrossentropy 而不是 CategoricalCrossentropy,因为 Torch 的交叉熵不像 Tensorflow 中的 CategoricalCrossentropy 那样使用一种热编码。

    【讨论】:

      猜你喜欢
      • 2020-01-13
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2013-07-27
      • 2014-12-15
      • 2019-10-01
      • 1970-01-01
      • 2018-06-21
      相关资源
      最近更新 更多