PyTorch基础——输入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维一般为样本数量。

你可能感兴趣的:(PyTorch,PyTorch,输入batch第0维)