PyTorch 张量数据类型

CODE_RabbitV

已于 2024-06-11 15:33:18 修改

阅读量1.2k

点赞数 23

分类专栏： PyTorch 学习文章标签： pytorch 人工智能 python

于 2024-06-11 14:51:05 首次发布

本文链接：https://blog.csdn.net/CODE_RabbitV/article/details/139592278

版权

【数据类型】Python 与 PyTorch 常见数据类型对应：
在这里插入图片描述
用 a.type() 获取数据类型，用 isinstance(a, 目标类型) 进行类型合法化检测

>>> import torch
>>> a = torch.randn(2,3)
>>> a
tensor([[-1.7818, -0.2472, -2.0684],
        [ 0.0117,  1.4698, -0.9359]])
>>> a.type()  ## 获取数据类型
'torch.FloatTensor'
>>> isinstance(a, torch.FloatTensor)  ## 类型合法化检测
True
>>>

【什么是张量】标量与张量：用 a.dim(), a.shape 或者 a.size() 查看 dim 为 0 是标量，否则是张量
```
>>> import torch
>>>
>>> a = torch.tensor(1) 
>>> a
tensor(1)
>>> a.dim()
>>> 0  ## 
```