pytorch中tensor的类型转换
(1)数据类型转换
在Tensor后加 .long(), .int(), .float(), .double()等即可,也可以用.to()函数进行转换,所有的Tensor类型可参考/docs/stable/
(2)数据存储位置转换
CPU张量 ----> GPU张量,使用()
GPU张量 ----> CPU张量,使用()
(3)与numpy数据类型转换
Tensor---->Numpy 使用 (),data为Tensor变量
Numpy ----> Tensor 使用 torch.from_numpy(data),data为numpy变量
(4)与Python数据类型转换
Tensor ----> 单个Python数据,使用(),data为Tensor变量且只能为包含单个数据
Tensor ----> Python list,使用(),data为Tensor变量,返回shape相同的可嵌套的list