如何给一个二维的Tensor添加一列或者一行呢
PyTorch的使用学习问题:如何给一个二维的Tensor添加一列或者一行呢?解答:用torch.unsqueeze和torch.cat例:import torchn = 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拓展成二维,
原创
2021-07-14 18:12:04 ·
14005 阅读 ·
0 评论