torch.unsqueeze
Posted jiangkejie
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch.unsqueeze相关的知识，希望对你有一定的参考价值。
unsqueeze在指定位置处插入尺寸为1的维度。
注意：返回的张量与此张量共享相同的基础数据。
Example:
>>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1, 2, 3, 4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
[ 2],
[ 3],
[ 4]])
以上是关于torch.unsqueeze的主要内容，如果未能解决你的问题，请参考以下文章
[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法
Pytorch中torch.unsqueeze()和torch.squeeze()函数解析