![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
学习Pytorch过程中的笔记
唯有一颗慎独心
渴望力量使我成为琦玉
展开
-
Jupyter 如何显示matplotlib的画图
使用Jupyter book实现感知机算法的时候,发现Jupyter并不会直接把matplotlib的画图显示出来,百度了一下,找到了方法。要先运行一句话:在运行自己的代码,就可以显示出来了,特别简单。...原创 2019-09-15 11:05:03 · 2626 阅读 · 1 评论 -
torch.max()、torch.sum()、A.unsqueeze()、torch.view()、x.permute()、x.transpose()
torch.max()举例:import torchx = torch.randn(4, 3)print(x)# 沿着行取最大值max_value, max_idx = torch.max(x, dim=1)# 每一行的最大值max_value# 每一行最大值的下标max_idx结果:tensor([[-0.1317, 0.1343, 0.9080]...原创 2019-10-13 11:52:34 · 1161 阅读 · 0 评论 -
Pytorch和Numpy的相互转换
PyTorch 的官方介绍是一个拥有强力GPU加速的张量和动态构建网络的库,其主要构件是张量,所以我们可以把 PyTorch 当做 NumPy 来用,PyTorch 的很多操作好 NumPy 都是类似的,但是因为其能够在 GPU 上运行,所以有着比 NumPy 快很多倍的速度。看一下两者之间的转换:import torchimport numpy as np# 创建一个...原创 2019-10-13 11:14:34 · 2086 阅读 · 0 评论 -
Pytorch多次自动求导
在Pytorch中,通过调用 backward 我们可以进行一次自动求导,如果我们再调用一次 backward,会发现程序报错,没有办法再做一次。这是因为 PyTorch 默认做完一次自动求导之后,计算图就被丢弃了,所以两次自动求导需要手动设置一个东西,我们通过下面的小例子来说明。import torchfrom torch.autograd import Variable...原创 2019-10-13 10:49:37 · 2170 阅读 · 2 评论 -
torch.randn()、torch.mean()、torch.pow()、torch.matmul()
torch.randn()产生大小为指定的,正态分布的采样点,数据类型是tensortorch.mean()torch.mean(input) 输出input 各个元素的的均值,不指定任何参数就是所有元素的算术平均值,指定参数可以计算每一行或者 每一列的算术平均数例如:a=torch.randn(3) #生成一个一维的矩阵b=torch.randn(1,3) #生成一个...原创 2019-10-13 10:44:34 · 15608 阅读 · 1 评论