Docs
at::Tensor cat(at::TensorList tensors, int64_t dim = 0i64) at::Tensor cat(at::TensorList tensors, at::Dimname dim)
두 번째 인자인 dim은 늘리고자 하는 차원을 의미한다. 파이토치에서 디폴트 값은 0이니 참고.
예제를 통해 이해하면 쉽다. 다음 링크를 참고: 8) 연결하기(concatenate)