[Pytorch] List를 Tensor로 변경하기. torch.stack

list에 tensor가 담겨져 있는 경우, list를 없애고 torch.tensor로 바꾸는 방법은 아래와 같다.

 

# segm_result = list[tensor]
segm_result = torch.stack(segm_result, dim=0)