Pytorch中的 torch.Tensor() 和 torch.tensor() 的区别

本文详细介绍了PyTorch中torch.Tensor和torch.tensor的区别。torch.Tensor是多维矩阵,主要用来创建32位浮点数张量,而torch.tensor()函数则根据传入的数据类型构造张量,若无数据则创建空张量,默认数据类型同样是32位浮点数。
部署运行你感兴趣的模型镜像

直接在搜索引擎里进行搜索,可以看到官方文档中两者对应的页面:

在这里插入图片描述
分别点击进去,第一个链接解释了什么是 torch.Tensor:

在这里插入图片描述

torch.Tensor 是一个包含单一数据类型元素的多维矩阵(数组)。

正因为 torch.Tensor 只包含单一的数据类型,所以后面文档就列出了不同数据类型对应的不同 Tensor。然后最后有一句话:

在这里插入图片描述

torch.Tensor 是默认张量类型 (torch.FloatTensor) 的别名

也就是说,torch.Tensor() 的作用实际上跟 torch.FloatTensor() 一样,都是生成一个数据类型为 32 位浮点数的张量,如果没传入数据就返回空张量,如果有列表或者 narray 的返回其对应张量。但无论传入数据本身的数据类型是什么,返回的都是 32 位浮点数的张量。

>>> torch.Tensor()
tensor([])
>>> torch.Tensor().dtype
torch.float32
>>> torch.FloatTensor()
tensor([])
>>> torch.FloatTensor().dtype
torch.float32

第二个链接,说明了 torch.tensor() 是一个通过数据构造张量的函数

在这里插入图片描述

重点是数据!!,这意味着 torch.tensor() 必须传入 data 参数,如果想要生成空张量,也要传入一个空列表,而这个空张量的数据类型是默认的 float32

>>> torch.tensor([])
tensor([])
>>> torch.tensor([]).dtype
torch.float32

如果是从已有的数据构造,那返回的张量的数据类型就跟原数据一致

>>> torch.tensor([1, 2, 3])
tensor([1, 2, 3])
>>> torch.tensor([1, 2, 3]).dtype
torch.int64

您可能感兴趣的与本文相关的镜像

PyTorch 2.7

PyTorch 2.7

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值