Fork me on GitHub 0
摘要: torchvision包功能是实现数据的处理、导入和预览等,通过包来获取手写数字的训练集和测试集。 import torch from torchvision import datasets,transforms from torch.autograd import Variable from to 阅读全文
posted @ 2020-04-07 22:04 amazingcode 阅读(167) 评论(0) 推荐(0) 编辑
摘要: 官方文档:https://docs.scipy.org/doc/numpy/reference/generated/numpy.transpose.html 对于二维 ndarray,transpose在不指定参数是默认是矩阵转置。如果指定参数,有如下相应结果: x.transpose((0,1)) 阅读全文
posted @ 2020-04-07 20:45 amazingcode 阅读(135) 评论(0) 推荐(0) 编辑
摘要: 在torchvision.transforms中常用的数据变换操作: torchvision.transforms.Resize:对图片数据按需求大小进行缩放,传递的参数为整型,(h,w)h表示高度,w表示高度 torchvision.transforms.Scale:对图片按需求大小进行缩放 to 阅读全文
posted @ 2020-04-07 18:45 amazingcode 阅读(1290) 评论(0) 推荐(0) 编辑
摘要: torch.autogard包的主要功能是完成神经网络后向传播中的链式求导。 实现自动梯度功能的过程大致为:先通过输入的Tensor数据类型的变量在神经网络的前向传播过程中生成一张计算图,然后根据这个计算图和输出结果准确计算出每个参数需要更新的梯度,并通过完成后向传播完成对参数的梯度更新。 Vari 阅读全文
posted @ 2020-04-07 18:21 amazingcode 阅读(286) 评论(0) 推荐(0) 编辑