When call to("cuda"), which one should be called

tgsmdww · June 16, 2020, 2:10am

We have to.dtype_layout, to.device, to.dtype, to.other. I use gdb to trace and find that it calls to.dtype_layout instead of to.device.

import torch

a = torch.Tensor([1,2,3]).to("cuda")

ptrblck · June 16, 2020, 8:24am

The to() method check for all input options. What kind of issue are you seeing?

tgsmdww · June 17, 2020, 2:19am

If I want to call to.dtype instead of to.dtype_layout, how to code?

ptrblck · June 17, 2020, 5:00am

If you want to change the dtype, you can directly pass the wanted dtype to the to() call:

x = x.to(torch.float32)
x = x.to(torch.long)

or call the “dtype method” on the tensor:

x = x.float()
x = x.long()

tgsmdww · June 17, 2020, 11:54am

Sorry, I’m wrong. When I use tensor.to(torch.long) it will call to.dtype. But When I call tensor.to("cuda") it will call to.dtype_layout instead of to.device.