脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch中Parameter函数

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 中 key 参数的含义及用法小结

    Python 中 key 参数的含义及用法小结

    我们在使用 sorted() 或 map() 函数的时候，都会看到里面有一个 key 参数，其实这个 key 参数也存在于其他内置函数中（例如 min()、max() 等），那么我们今天就来了解一下 key 参数的含义以及用途吧,需要的朋友可以参考下
    2023-12-12
  • Python利用PyVista进行mesh的色彩映射的实现

    Python利用PyVista进行mesh的色彩映射的实现

    这篇文章主要介绍了Python利用PyVista进行mesh的色彩映射的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-04-04
  • Python列表删除重复元素与图像相似度判断及删除实例代码

    Python列表删除重复元素与图像相似度判断及删除实例代码

    这篇文章主要给大家介绍了关于Python列表删除重复元素与图像相似度判断及删除的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-05-05
  • python 杀死自身进程的实现方法

    python 杀死自身进程的实现方法

    今天小编就为大家分享一篇python 杀死自身进程的实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • pyqt5自定义信号实例解析

    pyqt5自定义信号实例解析

    这篇文章主要介绍了pyqt5自定义信号实例解析，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下
    2018-01-01
  • Python print函数使用由浅入深全面详解

    Python print函数使用由浅入深全面详解

    这篇文章主要为大家介绍了Python print函数使用由浅入深全面详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12
  • Python自动扫码支付的实现示例

    Python自动扫码支付的实现示例

    本文主要介绍了Python自动扫码支付的实现示例,包括二维码识别、自动填写支付信息、自动点击支付等过程,具有一定的参考价值,感兴趣的可以了解一下
    2023-11-11
  • python中matplotlib实现最小二乘法拟合的过程详解

    python中matplotlib实现最小二乘法拟合的过程详解

    这篇文章主要给大家介绍了关于python中matplotlib实现最小二乘法拟合的相关资料，文中通过示例代码详细介绍了关于最小二乘法拟合直线和最小二乘法拟合曲线的实现过程，需要的朋友可以参考借鉴，下面来一起看看吧。
    2017-07-07
  • python三引号输出方法

    python三引号输出方法

    在本文里我们给大家分析那个了关于python三引号输出方法以及相关知识点，需要的朋友们学习下。
    2019-02-02
  • Python 多进程原理及实现

    Python 多进程原理及实现

    这篇文章主要介绍了Python 多进程原理及实现，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
    2020-12-12

最新评论