2020年3月27日
torchvision.transforms.ToTensor()，torchvision.trasnsforms.Normalize()
摘要： 用PyTorch进行神经网络训练时，如果训练用的数据是图像数据，则需要在训练之前对图像进行预处理。以MNIST数据为例： train_data = torchvision.datasets.MNIST( root='./mnist/', train=True, transform=torchvisi 阅读全文
posted @ 2020-03-27 22:47 Picassooo 阅读(3318) 评论(0) 推荐(0) 编辑
PyTorch: .add()和.add_()，.mul()和.mul_()，.exp()和.exp_()
摘要： .add()和.add_() .add()和.add_()都能把两个张量加起来，但.add_是in-place操作，比如x.add_(y)，x+y的结果会存储到原来的x中。Torch里面所有带"_"的操作，都是in-place的。 .mul()和.mul_() x.mul(y)或x.mul_(y)实 阅读全文
posted @ 2020-03-27 19:36 Picassooo 阅读(20234) 评论(0) 推荐(2) 编辑
PyTorch矩阵乘法
摘要： 总结：pytorch和numpy中，星号*都表示矩阵点对点相乘；matmul都表示矩阵乘法。 PyTorch中，a.mul(b)表示矩阵a和b点对点乘（即相同位置的元素相乘），‘*’也表示矩阵点对点相乘。a.mm(b)表示矩阵乘法。另外torch.matmul(a, b)也可以表示a和b的矩阵乘积。 阅读全文
posted @ 2020-03-27 11:22 Picassooo 阅读(1943) 评论(0) 推荐(0) 编辑