torch.tensor
infers the dtype
automatically, while torch.Tensor
returns a torch.FloatTensor
.
I would recommend to stick to torch.tensor
, which also has arguments like dtype
, if you would like to change the type.
16 Likes