[Pytorch] Tensor CPU, GPU 변경

 

sample_tensor.is_cuda # gpu 텐서 확인 리턴값 bool

sample_tensor.to(device) # 설정한 디바이스로 텐서 타입 변경
sample_tensor.cuda() # 다른 방식 1 
sample_tensor = sample_tensor.to(device. dtype=torch.float64) # 다른방식 2
sample_tensor = torch.as_tensor(sample_tensor, dtype=torch.float32, device='gpu') # 다른방식 3

sample_tensor.cpu() # CPU로 옮길 때