np.concatenate() 和 torch.cat()

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]