I want to stack list of something and convert it to gpu:
torch.stack(fatoms, 0).to(device=device)
As far as I know, tensor was created on cpu firstly and then would be transferred to specified device. How to put it on gpu straight?