PyTorch的使用学习
问题:
如何给一个二维的Tensor添加一列或者一行呢?

解答:
用torch.unsqueeze和torch.cat

例:

import torch
n = torch.randn([6,4])
p = torch.ones(n.shape[0])
p = p.unsqueeze(1)
print(n)
print(p)
n = torch.cat((n,p),1)
print(n)

因为不同维度的Tensor没有办法进行cat操作。
所以这里要用unsqueeze拓展成二维,再进行dim=1上的cat。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐