MNIST的均值和方差(0.1307,), (0.3081,)是怎么计算出来的?

0.1307和0.3081是mnist数据集的均值和标准差,因为mnist数据值都是灰度图,所以图像的通道数只有一个,因此均值和标准差各一个。

对于MNIST训练集,均值就是所有训练集(6000个样本)的像素点之和求平均。当你将训练样本从(6000,1,28,28)展开成(60000,28*28)后,用mean()函数可以轻松求出均值。同理用std()函数可以求出标准差

完整代码

import torch
from torchvision import transforms
from torchvision import datasets
from torch.utils.data import DataLoader
import torch.nn.functional as F
import torch.optim as optim

batch_size=60000 #这里是为了后面一次取出所有的数据
# transform=transforms.Compose([transforms.ToTensor

你可能感兴趣的:(pytorch,神经网络,深度学习,python)