I can use the ATen c++ library to create a tensor on the current device, e.g.
Tensor d = CUDA(kFloat).ones({3, 4})
Is it possible to create the tensor on a specific device.
i.e. to avoid calling cudaSetDevice(d), then create the tensor
Thanks
I can use the ATen c++ library to create a tensor on the current device, e.g.
Tensor d = CUDA(kFloat).ones({3, 4})
Is it possible to create the tensor on a specific device.
i.e. to avoid calling cudaSetDevice(d), then create the tensor
Thanks