【发布时间】：2021-03-16 11:56:08
【问题描述】：
我目前正在将 pytorch 代码翻译成 tensorflow。 有一点是我在 tensorflow 自定义循环中聚合了 3 个损失，我收到一个错误，我将二维数组与一维数组传递到 tensorflow 的 CategoricalCrossEntropy 中，这是非常合法的，我理解为什么会发生这种情况......但是在 pytorch 代码中，我传递了相同的形状，并且它与 CrossEntropyLoss 完美配合。有人知道我必须做什么才能将其转移到 TF 中吗？ 传入的形状是 (17000,100) vs (17000)
【问题讨论】：
标签： python numpy tensorflow pytorch