[libtorch] torch::cat 텐서 연결하기

gyeong·2021년 7월 23일
0

DeepLearning

목록 보기
1/4

PyTorch


Docs


libtorch STL


at::Tensor cat(at::TensorList tensors, int64_t dim = 0i64)
at::Tensor cat(at::TensorList tensors, at::Dimname dim)

두 번째 인자인 dim은 늘리고자 하는 차원을 의미한다.
파이토치에서 디폴트 값은 0이니 참고.

예제를 통해 이해하면 쉽다.
다음 링크를 참고: 8) 연결하기(concatenate)

profile
내가 보려고 만든 벨로그

0개의 댓글