conversión de tensor/tipo de datos de pytorch

¿Cómo convierte Pytorch los tipos de datos?

https://pytorch.org/docs/stable/generated/torch.Tensor.to.html

Usando métodos de clase to, puede convertir tipos de datos y posiciones de datos, es decir, puede convertir tensor antes que CPU y GPU.

tensor = torch.randn(2, 2)  # Initially dtype=float32, device=cpu
tensor.to(torch.float64)

cuda0 = torch.device('cuda:0')
tensor.to(cuda0)

tensor.to(cuda0, dtype=torch.float64)

other = torch.randn((), dtype=torch.float64, device=cuda0)
tensor.to(other, non_blocking=True)