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（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。