资讯动态

R torchvision 0.8.0 中 MNIST 张量缺少通道维的修复

发布时间:2026/10/27 2:42:07 来源:尧图企业网站定制
用 R 的torch+luz跑 MNIST 卷积神经网络时,训练一开始就崩在第一个卷积层,报错信息把 batch size 当成了通道数。这不是模型写错,而是数据加载器返回的张量形状变了。问题现象模型定义中卷积层的输入通道为 1:self$conv1=nn_conv2d(1,32,3)# N * 1 * 28 * 28 - N * 32 * 26 * 26调用fit()时报错:Error in (function (input, weight, bias, stride, padding, dilation, group

读完文章,也想定制专属网站?

尧图设计师 24 小时内与您沟通定制方案

免费获取报价 →
↑