PyTorch：nn.Module 中的批量大小和单个数据

在 pytorch nn.Module 中，创建的模型似乎与批量大小无关。也就是说，如果一个单独的数据是 128 个维度，并且我们以 64 个为一组进行训练，那么模型的输入应该是 128，而不是 128 x 64。

我的第一步nn.Sequential是一个Flatten. 当我将模型应用于单个基准（无批次）时，我需要确保 Flatten 具有start_dim=0. 但这在应用于批次时是不正确的。这似乎是与上面相反的界面：您需要根据是否使用批处理来定制模型。

所以：

1. 是否nn.Module需要注意批处理？
2. 如果是：如何将模型应用于单个样品，而不是批次？
3. 如果不是：您如何应用 Flatten，您可能会发送一批，或者您可能会发送一个样品？

一个等效的问题可能是：我如何构建一个 PyTorch 模型来批量训练，但仍然在生产时将其应用于单个数据？