根据Deep Learning with PyTorch英文原版
Chapter 6 Using a neural network to fit the data (Page 154)中:
To accommodate multiple samples, modules expect the zeroth dimension of the input to be the number of samples in the batch.
PyToch中输入batch第0维一般为样本数量。
根据Deep Learning with PyTorch英文原版
Chapter 6 Using a neural network to fit the data (Page 154)中:
To accommodate multiple samples, modules expect the zeroth dimension of the input to be the number of samples in the batch.
PyToch中输入batch第0维一般为样本数量。