torch.unsqueeze()
torch.unsqueeze()
返回在指定位置插入的尺寸为1的新张量
dim-插入的单个尺寸的索引
import torch
X=torch.tensor([1,2,3,4])
X.size()
torch.Size([4])
X1=torch.unsqueeze(X,0) # 在第0维上增加“1”,在第一个维度上增加一个维度
X1
tensor([[1, 2, 3, 4]])
X1.size()
torch.Size([1, 4])
参考博客
...
原创
2021-05-14 11:05:18 ·
102 阅读 ·
0 评论