torch.cat函数

pytorch中cat和concat函数完全一样，没有任何区别！！！

however，mxnet中的nd.concat的dim参数如果不指定，默认就是左右拼接，和pytorch中的dim不指定就是上下拼接完全相反！！

#dim=0,上下拼接
#dim=1，左右拼接

import torch
# x1
x1 = torch.tensor([[1,2,3],[4,5,6]],dtype=torch.int)

# x2
x2 = torch.tensor([[7,8,9],[10,11,12]],dtype=torch.int)


#'inputs为２个形状为[2 , 3]的矩阵 '
inputs = [x1, x2]
print(inputs)


#dim=0,上下拼接
print(torch.cat(inputs, dim=0))
#dim=1，左右拼接
print(torch.cat(inputs, dim=1))