Tensor创建:依据数值创建

本文详细介绍如何使用PyTorch创建各类张量,包括全0张量、全1张量、自定义数值张量、等差张量、均分数列及单位对角矩阵。涵盖torch.zeros(), torch.ones(), torch.full(), torch.arange(), torch.linspace() 和 torch.eye() 的用法。

Tensor创建:依据数值创建

3.Tensor:依据数值创建

(1)创建全0张量

torch.zeros()

有五个属性
size:张量的形状 如(3,224,224)
out:输出张量
layout:内存中的布局形式,默认stride,但当存储矩阵为稀疏矩阵时可设为 sparse_coo,提高查询效率等
device:所在设备
requirs_grad:是否需要梯度

zeros=torch.zeros((3,3),out=out_t)

zeros与out_t也是共享同一内存空间的
在这里插入图片描述
在这里插入图片描述
(2)torch.zeros_like()
依据input形状创建全0张量
在这里插入图片描述
(3)全1张量
torch.ones()
(4)torch.ones_like()

(5)自定义数值的张量
torch.full(size,fill_value)

torch.full((6,6),666)

在这里插入图片描述

(6)torch.full_like()

(7)等差张量
torch.arange()
功能:创建等差一维张量
数值区间为{start,end) end取不到
step 为公差,默认1
在这里插入图片描述

t=torch.arange(2,10,2)
t

在这里插入图片描述
(8)均分数列

torch.linspace()
创建一维均分数列张量
区间为闭合的区间 [start,end]
在这里插入图片描述

t=torch.linspace(0,10,5)
t

在这里插入图片描述

(10)创建单位对角矩阵(二维张量)
torch.eye()
默认为方阵
n:为行数
m:为列数
在这里插入图片描述

torch.eye(10)

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值