- 📢博客主页:盾山狂热粉的博客_优快云博客-C、C++语言,机器视觉领域博主
- 📢努力努力再努力嗷~~~✨
💡大纲
⭕首先认识pytorch的基本数据类型:张量tensor
| 类型 | ||||||||
|---|---|---|---|---|---|---|---|---|
| CPU | torch.FloatTensor | torch.DoubleTensor | torch.HalfTensor | torch.ByteTensor | torch.CharTensor | torch.ShortTensor | torch.IntTensor | torch.LongTensor |
| GPU | 例如torch.cuda.FloatTensor | |||||||
一、查看张量状态
(一)引用模块
import torch
(二)对不同维度的张量进行访问
1、Dimension 0
💡一般是一个标量,维度为0
👉取tensor里面的值
a = torch.tensor(1)
print(a) # tensor(1)
print(a.item()) # 1
⚠️.items() 这是字典里取键值对
👉输出当前类型
print(a.type()) # 这种信息更详细
print(type(a))
'''
torch.LongTensor
<class 'torch.Tensor'>
'''

最低0.47元/天 解锁文章
779

被折叠的 条评论
为什么被折叠?



