pytorch: Parameter 的数据结构实例
一般来说,pytorch的Parameter是一个tensor,但是跟通常意义上的tensor有些不一样
1)通常意义上的tensor仅仅是数据
2)而Parameter所对应的tensor除了包含数据之外,还包含一个属性:requires_grad(=True/False)
在Parameter所对应的tensor中获取纯数据,可以通过以下操作:
param_data=Parameter.data
测试代码:
#-*-coding:utf-8-*- importtorch importtorch.nnasnn ##regressionforthe3*2affinematrix fc_loc=nn.Sequential( nn.Linear(10*3*3,32), nn.ReLU(True), nn.Linear(32,3*2) ) ##initializetheweights/biaswithidentytransformation 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的数据结构实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。