np.concatenate() 和 torch.cat()

原创于 2018-06-28 14:32:56 发布 · 1.6w 阅读

12 ·

CC 4.0 BY-SA版权

PyTorch 专栏收录该内容

25 篇文章

订阅专栏

本文介绍如何使用PyTorch进行张量的拼接操作，并给出具体示例代码，包括将两个形状为[1,3,256,256]的张量沿通道维度拼接成一个[1,6,256,256]的张量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

self._A=np.array(self.real_A) #[1,3,256,256]
self._B=np.array(self.real_B) #[1,3,256,256]
input=np.concatenate((self._A, self._B),axi=1)
input=torch.from_numpy(input) #[1,6,256,256]
等同于 
input=torch.cat((self.real_A,self.real_B),dim=1) 
# print 'input',input.size() [1,6,256,256]