欢迎光临散文网 会员登陆 & 注册

01-02-张量简介与创建

2022-07-25 21:18 作者:冰糖炖子瑜  | 我要投稿

张量——是一个多维数组,它是标量、向量、矩阵的高维拓展

张量创建

一、直接创建

data可以是list、numpy

notice:torch.from_numpy 共享内存

二、依据数值创建

torch.zeros()

torch.zeros_like()

torch.ones()

torch.ones_like()

torch.full()

torch.full_like()

torch.arrange() 创建等差的一维张量

torch.linspace() 创建均分的一维张量

steps 数列长度,即张量元素的个数

torch.logspace() 创建对数均分的一维张量

base为对数的底

torch.eye() 创建单位对角矩阵(2维张量)

三、依概率分布创建张量

torch.nomal() 生成正态分布(高斯分布)

标准差:


torch.nomal()四种模式

torch.randn() 生成标准正态分布

torch.randn_like()

torch.rand() 在区间[0,1]上生成均匀分布

torch.rand_like()

torch.randint() 在区间[low,high]上,生成整数均匀分布

torch.randint_like()

torch.randperm()

torch.bernoulli()



01-02-张量简介与创建的评论 (共 条)

分享到微博请遵守国家法律