PyTorch基礎——輸入batch第0維含義

gy笨瓜發表於2020-10-12

根據Deep Learning with PyTorch英文原版
Chapter 6 Using 啊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維一般為樣本數量。

相關文章