pytorch: Parameter 的数据结构实例

yipeiwu_com6年前 (2020-03-06)Python基础

一般来说,pytorch 的Parameter是一个tensor,但是跟通常意义上的tensor有些不一样

1) 通常意义上的tensor 仅仅是数据

2) 而Parameter所对应的tensor 除了包含数据之外,还包含一个属性:requires_grad(=True/False)

在Parameter所对应的tensor中获取纯数据,可以通过以下操作:

param_data = Parameter.data

测试代码:

#-*-coding:utf-8-*-
import torch
import torch.nn as nn
 
## regression for the 3 * 2 affine matrix
fc_loc = nn.Sequential(
  nn.Linear(10 * 3 * 3, 32),
  nn.ReLU(True),
  nn.Linear(32, 3 * 2)
)
 
## initialize the weights/bias with identy transformation
fc_loc[2].weight.data.zero_()
fc_loc[2].bias.data.copy_(torch.tensor([1, 0, 0, 0, 1, 0], dtype=torch.float))
# print(fc_loc)
print(fc_loc[2].weight)
print(fc_loc[2].weight.data)

以上这篇pytorch: Parameter 的数据结构实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。

返回列表

上一篇：Python字符转换

下一篇：PHP生成静态页面详解

相关文章

Python实现打砖块小游戏代码实例

Python实现打砖块小游戏代码实例

这次用Python实现的是一个接球打砖块的小游戏，需要导入pygame模块，有以下两条经验总结： 1.多父类的继承2.碰撞检测的数学模型知识点稍后再说，我们先看看游戏的效果和实现：一...

django manage.py扩展自定义命令方法

django manage.py扩展自定义命令方法

# django manage.py扩展自定义命令环境： mac django1.10.3 在实际的项目开发过程中，我们可能要执行某脚本初始化数据库，可能要启动多个服务，比...

$python: line=f.readlines()消除line中\n的方法$

python: line=f.readlines()消除line中\n的方法

测试代码 #!/ust/bin/env python3 f = open("name.txt") date = f.readlines() print(date) f.close(...

Python快速排序算法实例分析

Python快速排序算法实例分析

本文实例讲述了Python快速排序算法。分享给大家供大家参考，具体如下：快速排序的时间复杂度是O(NlogN) 算法描述： ① 先从序列中取出一个数作为基准数 ② 分区过程, 将比这个...

使用Python导出Excel图表以及导出为图片的方法

使用Python导出Excel图表以及导出为图片的方法

本篇讲下如何使用纯python代码将excel 中的图表导出为图片。这里需要使用的模块有win32com、pythoncom模块。网上经查询有人已经写好的模块pyxlchart，具体代...