深度学习框架pytorch1

• 📢博客主页：盾山狂热粉的博客_优快云博客-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'>
'''

👉查看设备（CPU、GPU）

print(a.device) # cpu