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

被折叠的 条评论
为什么被折叠?



