Add int and str type to device parameter of torch.tensor()

[hyperkai]

📚 The doc issue

The doc explains device parameter but it doesn't have int and str type as shown below:

device (torch.device, optional) – the device of the constructed tensor. If None and data is a tensor then the device of data is used. If None and data is not a tensor then the result tensor is constructed on the current device.

Actually, device parameter with int and str type works as shown below:

import torch

torch.tensor(data=-3, device=0)
torch.tensor(data=-3, device='cuda:0')
# tensor(-3, device='cuda:0')

Suggest a potential alternative/fix

No response

cc @albanD

[albanD]

Adding string sounds definitely good.
I'm not so sure about int as I'm not sure if it is not something we would like to change going forward.