Pytorch————学习1

torch.nn  仅支持小批量。整个torch.nn程序包仅支持作为小批量样本的输入,而不支持单个样本。

例如,nn.Conv2d采用的是4D张量:nSamples nChannels Height Width

posted @ 2020-11-24 21:00  cfancy  阅读(43)  评论(0编辑  收藏  举报