주어진 차원에 맞춰 텐서(Tensor)를 어떻게 합칠 수 있나요?

자주 묻는 질문&답변
1

공식 홈페이지와 StackOverflow 등에서 자주 보이는 질문과 답변을 번역하고 있습니다.

다음 링크에서 원문을 함께 찾아보실 수 있습니다.

질문

  • Tensorflow에서는 다음과 같이 텐서를 합칠 수 있었습니다.
    third_tensor= tf.concat(0, [first_tensor, second_tensor])
  • 이 때, first_tensor와 second_tensor의 크기(shape)가 각각 [5, 32, 32]일 때,
    첫번째 차원(dimension)은 배치 크기(batch_size)입니다.
    이렇게 두 개의 텐서를 합친 third_tensor의 크기는 [10, 32, 32]가 됩니다.
  • 파이토치에서는 어떻게 할 수 있나요?
2

답변

  • torch.cat()의 문서를 참고하세요.
  • 예를 들어 아래와 같이 할 수 있습니다.
    third_tensor = torch.cat((first_tensor, second_tensor), 0)

더 알아보기