list 안에 이미 torch.Tensor이 있고, 그것들의 dtype이 CUDA일 경우, 이를 하나의 torch.Tensor로 변환하려면 torch.stack을 사용할 수 있습니다. 이 함수는 주어진 텐서의 리스트를 하나의 텐서로 쌓아줍니다.

다음은 예제 코드입니다:

python
Copy code
import torch

가상의 리스트

tensor_list = [torch.randn(3, 3).cuda() for _ in range(5)]

리스트의 텐서를 하나로 쌓기

stacked_tensor = torch.stack(tensor_list)

print(stacked_tensor)
print(stacked_tensor.shape)
print(stacked_tensor.type())

이 코드에서 torch.randn(3, 3).cuda()는 가상의 CUDA 텐서를 생성하고, 이를 리스트에 추가한 후 torch.stack을 사용하여 리스트 안의 텐서를 하나로 쌓아줍니다. 결과적으로 stacked_tensor는 하나의 텐서로 변환됩니다.