pytorch torch.cat((A, B), 2)