tensor 新建 --pytorch

tensor 新建 --pytorch

1. 新建tensor的方法

1.1 import from numpy

在这里插入图片描述

1.2 import from list

在这里插入图片描述

1.3 uninitalized

在这里插入图片描述

使用前需要初始化,不然,容易nan,inf

1.4 **set default type **

在这里插入图片描述

这个是设置tensor的类型,只能设置float 或者double,其他的不行

1.5 **rand/ rand_like, randinit **

在这里插入图片描述

1.6 randn

这个是正态分布的
N(0,1),均值为0,方差为1

在这里插入图片描述

用normal的话,得注意后面std的参数,std=torch.arange(0,0.9,0.1) 他也是左闭右开的区间,标准差从最小是0,最大是0.9 [0, 0.9) ,但是这个间隔要和前面的full的size一致,看错误提示就知道了

1.7 full

全部都是啥
标量的话 full([],x),写full([0],x), 0维张量不是一个东东

在这里插入图片描述

1.8 arange/range

range 不用了,剔除了 [0,10)

在这里插入图片描述

1.9 linspace/logspace

等差,等比数列 linspace(start, end, steps) logspace(start, end, steps=100,base=10)
这个是[0, 10], 左闭右闭区间
在这里插入图片描述

1.10 zeros/ones/eye

全是0, 1, 对角为1

在这里插入图片描述
在这里插入图片描述
xxx_like, 就是不用输入shape,直接复制对象的shape

1.11 randperm

随机打散
numpy要两步,torch 一下子

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

发布了65 篇原创文章 · 获赞 7 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_36628778/article/details/103972548