2020年2月13日

pytorch(3)----基本类型Autograd计算图

摘要： pytorch 0.4版本之后，torch.autograd.Variable 和torch.Tensor 进行了整合。 基本内容 1、 创建可以自动求导的Tensor，默认为false。requires_grad属性 2、 Tensor 的连个属性： grad: 记录该Tensor对应的梯度 gr 阅读全文

posted @ 2020-02-13 22:40 feihu_h 阅读(318) 评论(0) 推荐(0) 编辑

pytorch(2)----基本数据类型与模块

摘要： Tensor 张量，包含单一数据类型元素的矩阵。 基本内容 1、数据初始化及数据类型转换 2、组合： torch.cat() 按照某一个维度进行拼接，总维度数目不变 torch.stack() 按照制定维度进行叠加，新增维度 3、分块：torch.chunk() 指定分块数量 torch.split 阅读全文

posted @ 2020-02-13 19:09 feihu_h 阅读(409) 评论(0) 推荐(0) 编辑

导航