1.显示pytorch中tensor的数据类型:
import torch
x=torch.Tensor([1,2])
print('x: ',x)
print('type(x): ',x.size() ) # 查看tensor的维度是什么样,查看tensor的shape
print('x.dtype: ',x.dtype) # 这个才是查看tensor中数据的具体类型是什么
pytorch: 获取 tensor 维度的方法:shape 和 size()_Caesar6666的博客-CSDN博客_查看tensor的shape
2. 如何转换tensor中数据的数据类型
x = torch.tensor(x, dtype=torch.float64)
将数据类型转换成:float64类型。
会提示一个warning : UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor).
但是不用管,,,
3,一些疑惑待解决:
PS:但有的教程给出了还要将x = variable(x) 的方式,不是很明白。但是链接放在这: