-
train/eval过程中,打印loss信息
data.item() ----> Tensor --> float -
数据类型转换
在Tensor后加 .long(), .int(), .float(), .double()等即可,也可以用.to()函数进行转换, Tensor类型官网 -
数据存储位置转换
CPU张量 ----> GPU张量,使用data.cuda()
GPU张量 ----> CPU张量,使用data.cpu() -
与numpy数据类型转换
Tensor---->Numpy 使用 data.numpy(),data为Tensor变量
Numpy ----> Tensor 使用 torch.from_numpy(data),data为numpy变量 -
与Python数据类型转换
Tensor ----> 单个Python数据,使用data.item(),data为Tensor变量且只能为包含单个数据
Tensor ----> Python list,使用data.tolist(),data为Tensor变量,返回shape相同的可嵌套的list
Pytorch中Tensor的类型转换
最新推荐文章于 2024-02-24 16:45:07 发布