摘要: torch.nn 仅支持小批量。整个torch.nn程序包仅支持作为小批量样本的输入,而不支持单个样本。 例如,nn.Conv2d采用的是4D张量:nSamples x nChannels x Height x Width 阅读全文
posted @ 2020-11-24 21:00 cfancy 阅读(43) 评论(0) 推荐(0) 编辑