如何给一个二维的Tensor添加一列或者一行呢

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。