pytorch中Parameter函数用法示例
更新时间：2022年01月30日 10:38:42 作者：鬼道2022
这篇文章主要为大家介绍了pytorch中Parameter函数用法，并用详细的代码示例进行演示详解，有需要的朋友可以借鉴参考下，希望能够有所帮助
用法介绍
pytorch中的Parameter函数可以对某个张量进行参数化。它可以将不可训练的张量转化为可训练的参数类型，同时将转化后的张量绑定到模型可训练参数的列表中，当更新模型的参数时一并将其更新。
torch.nn.parameter.Parameter
- data (Tensor)：表示需要参数化的张量
- requires_grad (bool, optional)：表示是否该张量是否需要梯度，默认值为True
代码介绍
pytorch中的Parameter函数具体的代码示例如下所示
import torch import torch.nn as nn class NeuralNetwork(nn.Module): def __init__(self, input_dim, output_dim): super(NeuralNetwork, self).__init__() self.linear = nn.Linear(input_dim, output_dim) self.linear.weight = torch.nn.Parameter(torch.zeros(input_dim, output_dim)) self.linear.bias = torch.nn.Parameter(torch.ones(output_dim)) def forward(self, input_array): output = self.linear(input_array) return output if __name__ == '__main__': net = NeuralNetwork(4, 6) for param in net.parameters(): print(param)
代码的结果如下所示：
当神经网络的参数不是用Parameter函数参数化直接赋值给权重参数时，则会报错，具体的程序
import torch import torch.nn as nn class NeuralNetwork(nn.Module): def __init__(self, input_dim, output_dim): super(NeuralNetwork, self).__init__() self.linear = nn.Linear(input_dim, output_dim) self.linear.weight = torch.zeros(input_dim, output_dim) self.linear.bias = torch.ones(output_dim) def forward(self, input_array): output = self.linear(input_array) return output if __name__ == '__main__': net = NeuralNetwork(4, 6) for param in net.parameters(): print(param)
代码运行报错结果如下所示:
以上就是pytorch中Parameter函数用法示例的详细内容，更多关于pytorch中Parameter函数的资料请关注脚本之家其它相关文章！
相关文章
Python sklearn中的.fit与.predict的用法说明
这篇文章主要介绍了Python sklearn中的.fit与.predict的用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要为大家详细介绍了基于python实现简单日历，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-07-07
Python 中urls.py:URL dispatcher（路由配置文件）详解
这篇文章主要介绍了Python 中urls.py:URL dispatcher（路由配置文件）详解的相关资料,需要的朋友可以参考下2017-03-03
这篇文章主要介绍了Python操作MySQL简单实现方法,通过一个简单的实例讲述了Python针对mysql数据库的增删改查技巧,需要的朋友可以参考下2015-01-01
tkinter其实是Python调用tcl程序的标准Python程序，可以通过这个interface调用tcl的程序，因为在大多数的unix系统中都内置了很多的tcl程序和命令。这篇文章主要介绍了python tkinter模块安装,需要的朋友可以参考下2020-01-01
这篇文章主要介绍了python开发之文件操作用法,以实例形式较为详细的分析了Python针对文件的路径、文件名、后缀名等操作技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-11-11
这篇文章主要介绍了python实现的简单猜数字游戏,涉及Python操作随机数的技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-04-04
栈和队列是在程序设计中常见的数据类型，从数据结构的角度来讲，栈和队列也是线性表，是操作受限的线性表。本文将详细介绍一下Python中的栈，感兴趣的可以了解一下2022-03-03![python 中的[:-1]和[::-1]的具体使用](//img.jbzj.com/images/xgimg/bcimg8.png)
这篇文章主要介绍了python 中的[:-1]和[::-1]的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-02-02
浅谈Django自定义模板标签template_tags的用处
这篇文章主要介绍了浅谈Django自定义模板标签template_tags的用处，具有一定借鉴价值，需要的朋友可以参考下。2017-12-12
![python 中的[:-1]和[::-1]的具体使用](http://img.jbzj.com/images/xgimg/bcimg8.png)
最新评论