文章目录
一、unsqueeze() 方法
unsqueeze() 方法起升维的作用，参数表示在哪个地方加一个维度。
二、实例
import torch
input=torch.arange(0,6)
print(input)
print(input.shape)
print(input.unsqueeze(0)) # 在0维处加一维
print(input.unsqueeze(0).shape)
print(input.unsqueeze(1)) # 在1维处加一维
print(input.unsqueeze(1).shape)
输出结果：
tensor([0, 1, 2, 3, 4, 5])
torch.Size([6])
tensor([[0, 1, 2, 3, 4, 5]])
torch.Size([1, 6]) # 在0维处加了一维
tensor([[0],
[1],
[2],
[3],
[4],
[5]])
torch.Size([6, 1]) # 在1维处加了一维
参考链接
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/118861.html
