torch.nn 仅支持小批量。整个torch.nn程序包仅支持作为小批量样本的输入，而不支持单个样本。
例如，nn.Conv2d采用的是4D张量：nSamples x nChannels x Height x Width