如何解决keras 中 MSE 损失计算中的轴是什么?
我正在训练一个 CNN 来解决回归问题,我在带有 keras 后端的 tensorflow 中使用 channel_first 实现。
我的预测中有垂直条纹,我认为这与损失函数有关。我刚刚意识到 keras 使用 axis = -1
来进行 MSE 损失。
这个轴是什么意思?考虑我有一个 (batch,1,col,row) 张量并且我使用 MSE,这是否意味着它只计算我的行 (axis = -1
)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。