开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >专栏 >Expected input batch_size (1) to match target batch_size (4)：先找模型里丢失的 batch 维

Expected input batch_size (1) to match target batch_size (4)：先找模型里丢失的 batch 维

用户12833544

发布于 2026-11-14 14:48:24

发布于 2026-11-14 14:48:24

60

举报

概述

原帖 DataLoader batch_size=4，但交叉熵收到的输出第一维为 1、目标第一维为 4；贴出的代码没有 forward：

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。