pytorch中tensor如何转换int和float数据格式，如float32、float64、int32、int64

解决方法：使用torch.tensor(x, dtype=数据格式)，修改dtype即可

import torch

a = [1, 2, 3, 4]
a1 = torch.tensor(a)
a_float = torch.tensor(a1, dtype=torch.float32)
a_int64 = torch.tensor(a1, dtype=torch.int64)

print(a1.dtype)
print(a_float.dtype)
print(a_int64.dtype)

输出：

torch.int64
torch.float32
torch.int64