python函数集合学习ing

最新推荐文章于 2024-03-01 20:53:52 发布

wuling129

最新推荐文章于 2024-03-01 20:53:52 发布

阅读量133

点赞数

分类专栏：图像处理+python+深度学习文章标签： python

本文链接：https://blog.csdn.net/wuling129/article/details/118001961

版权

图像处理+python+深度学习专栏收录该内容

113 篇文章 10 订阅

订阅专栏

PyTorch 中文教程

1.torch.nn.Parameter()

PyTorch里面的torch.nn.Parameter()

PyTorch里面的torch.nn.Parameter() 简书

用法self.v = torch.nn.Parameter(torch.FloatTensor(hidden_size)),看了官方教程里面的解释也是云里雾里，于是在栈溢网看到了一篇解释。首先可以把这个函数理解为类型转换函数，将一个不可训练的类型Tensor转换成可以训练的类型parameter并将这个parameter绑定到这个module里面(net.parameter()中就有这个绑定的parameter，所以在参数优化的时候可以进行优化的)，所以经过类型转换这个self.v变成了模型的一部分，成为了模型中根据训练可以改动的参数了。使用这个函数的目的也是想让某些变量在学习的过程中不断的修改其值以达到最优化。

2、torch.pow()

torch.pow()的使用举例

功能: 实现张量和标量之间逐元素求指数操作, 或者在可广播的张量之间逐元素求指数操作.

3、torch.div()

torch.div()的使用举例

功能：张量和标量做逐元素除法或者两个可广播的张量之间做逐元素除法

4、torch.sum()

torch.sum()的用法

功能：对输入的tensor数据的某一维度求和，一共两种用法

１．torch.sum(input, dtype=None)
２．torch.sum(input, list: dim, bool: keepdim=False, dtype=None) → Tensor
　
input:输入一个tensor
dim:要求和的维度，可以是一个列表
keepdim:求和之后这个dim的元素个数为１，所以要被去掉，如果要保留这个维度，则应当keepdim=True
#If keepdim is True, the output tensor is of the same size as input except in the dimension(s) dim where it is of size 1.

import torch

a = torch.ones((2, 3))
a1 =  torch.sum(a)
a2 =  torch.sum(a, dim=0)
a3 =  torch.sum(a, dim=1)

print(a)
print(a1)
print(a2)
print(a3)
输出：