torch-tensor

torch.Tensor는 PyTorch의 다차원 배열 자료구조로서 CPU와 CUDA 메모리 모두에서 존재할 수 있고 자동미분을 위한 grad 정보를 포함할 수 있다. 이 확장에서는 CPU/CUDA 텐서와 grad 유무를 감지해 이미지 변환을 수행하며 텐서를 이미지 픽셀 범위로 정규화해 시각화한다. 텐서의 차원과 dtype을 확인해 색 채널과 배치 처리를 결정하는 점이 시각화 정확도에 영향을 준다.