拼接tensor code [torch.cat(sele_nei, dim=0)]

静静_jingjing

于 2021-08-04 23:13:32 发布

阅读量143

点赞数

分类专栏： python编程文章标签： python

本文链接：https://blog.csdn.net/qq_39321513/article/details/119395383

版权

python编程专栏收录该内容

36 篇文章 1 订阅

订阅专栏

本文通过示例展示了如何使用PyTorch创建张量，并进行张量的append及concat操作。首先定义了三个一维张量a、b、c，接着将它们添加到列表sele_nei中，并使用torch.cat进行拼接。此过程有助于理解PyTorch中张量的基本操作。

摘要由CSDN通过智能技术生成

import torch
sele_nei = []
a = torch.tensor([2, 3, 6, 0, 1])[np.newaxis]
b = torch.tensor([4, 3, 6, 0, 1])[np.newaxis]
c = torch.tensor([2, 3, 4, 0, 1])[np.newaxis]
print("a: ", a)
sele_nei.append(a)
sele_nei.append(b)
sele_nei.append(c)
print("sele_nei:  ", sele_nei)
sele_nei = torch.cat(sele_nei, dim=0)
print("sele_nei:  ", sele_nei)