Pytorch————学习1

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

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

原文地址:https://www.cnblogs.com/cfancy/p/14032628.html