
pytorch
shiyueyue0822
这个作者很懒,什么都没留下…
展开
-
pytorch中DataLoader使用torchvision.transforms.Lambda进行图片变换时的问题
pytorch中DataLoader使用torchvision.transforms.Lambda进行图片变换时的问题 将lambda方法放在外面实现 # -*- coding: utf-8 -*- """ # @file name : train_lenet.py # @author : tingsongyu # @date : 2019-09-07 10:08:00 # @brief : 人民币分类模型训练 """ import os import random im原创 2020-08-06 18:55:35 · 982 阅读 · 0 评论 -
张量
张量是多维数组,是标量、向量、矩阵的高维拓展 数据类型 Pytorch0.4.0之前的张量数据类型:torch.autograd.Variable, 用处:用来封装Tensor,进行自动求导 属性: data-----被包装的Tensor grad-----data的梯度 grad_fn:创建Tensor的Function,是自动求导的关键 requires_grad:指示是否需要梯度 Is_leaf:指示是否是叶子结点(张量) ====================...原创 2020-08-04 19:41:35 · 264 阅读 · 0 评论