![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
yuanCruise
工科生的笔尖
展开
-
pytorch中的参数类——torch.nn.parameter
1.torch.nn.parameter概要pytorch官网对torch.nn.parameter的描述如下。torch.nn.parameter是一个被用作神经网络模块参数的tensor。这是一种tensor的子类。parameters是张量的子类,当与模块s一起使用时,它们有一个非常特殊的属性——当它们被重新分配为模块属性时,它们会自动地添加到它的参数列表中,并且会出现在Parame...原创 2020-02-13 10:59:52 · 4231 阅读 · 1 评论 -
pytorch中的神经网络子模块(线性模块)——torch.nn.Linear
pytorch中的线性模块的实现如下,在init函数中定义weight值和bias值。class Linear(Module): __constants__ = ['bias', 'in_features', 'out_features'] def __init__(self, in_features, out_features, bias=True): sup...原创 2020-02-13 10:59:15 · 1995 阅读 · 2 评论 -
pytorch中的神经网络模块基础类——torch.nn.Module
1.torch.nn.Module概要pytorch官网对torch.nn.Module的描述如下。torch.nn.Module是所有的神经网络模块的基类,且所有的神经网络模块都可以包含其他的子神经网络模块,这些子神经网络模块可以作为类的属性赋值。2.torch.nn.Module.parameterspytorch官网对torch.nn.Module.parameters的描述如下:...原创 2020-02-13 10:57:52 · 976 阅读 · 0 评论 -
pytorch中的顺序容器——torch.nn.Sequential
1.torch.nn.Sequential概要pytorch官网对torch.nn.Sequential的描述如下。使用方式:# 写法一net = nn.Sequential( nn.Linear(num_inputs, 1) # 此处还可以传入其他层 )# 写法二net = nn.Sequential()net.add_module('linear', ...原创 2020-02-10 21:02:49 · 1668 阅读 · 0 评论 -
pytorch | 文档学习_持续更新
1.自动求导机制每一个tensor都有一个required_grad字段,该字段能够从梯度计算中以最小计算为单位对计算图进行操作,更加灵活效率更高。某个tensor的来源中,只要有一个tensor的required_grad字段为True,则当前tensor的required_grad为True。...原创 2019-12-18 19:28:29 · 299 阅读 · 0 评论