pytorch中unsqueeze_（增加维度）用法

最新推荐文章于 2024-09-27 15:27:34 发布

燕小硕

最新推荐文章于 2024-09-27 15:27:34 发布

阅读量6.9k

点赞数 2

文章标签： python 深度学习人工智能 pytorch 神经网络

本文链接：https://blog.csdn.net/weixin_44928646/article/details/104849988

版权

unsqueeze_是pytorch中升高维度的方法，numpy中不能用
unsqueeze_（0）：在0轴上加一个维度
unsqueeze_（1）：在1轴上加一个维度
。
。
。

import torch
from PIL import Image
import numpy as np
from torchvision import transforms
img=Image.open(r'C:\Users\Administrator\Pictures\QQ浏览器截图\QQ浏览器截图20190301211443.png')
a=np.array(img)
b=torch.Tensor(a)
print(b.shape)
#方法1：unsqueeze_
b.unsqueeze_(0)
print(b.shape)

在这里插入图片描述

import torch
from PIL import Image
import numpy as np
from torchvision import transforms
img=Image.open(r'C:\Users\Administrator\Pictures\QQ浏览器截图\QQ浏览器截图20190301211443.png')
a=np.array(img)
print(a.shape)
#方法二：array变成tensor的时候，加一个[]，可以直接加一个维度
b=torch.Tensor([a])
print(b.shape)