tensor中数据类型的相互转换

tensor中数据类型的相互转换,第1张

Pytorch中的Tensor常用的类型转换函数

    tensor数据类型转换
    在tensor的后面添加: .int()、.float()、.double()等.
    同时也可以使用 .to(type) 进行实现。
    同时也可以使用 type()函数 ,data为Tensor数据类型,data.type()为给出data的类型,如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。当你不知道要转换为什么类型时,但需要求a1,a2两个张量的乘积,可以使用a1.type_as(a2)将a1转换为a2同类型。
    tensor<–>numpy
    Tensor---->Numpy 使用 tensor.numpy(),tensor为Tensor变量
    Numpy ----> Tensor 使用 torch.from_numpy(data),data为numpy变量
    tensor<–>Python
    Tensor ----> 单个Python数据,使用data.item(),data为Tensor变量且只能为包含单个数据
    Tensor ----> Python list,使用data.tolist(),data为Tensor变量,返回shape相同的可嵌套的list
    数据存储位置转换
    CPU张量 ----> GPU张量,使用data.cuda()
    GPU张量 ----> CPU张量,使用data.cpu()
 

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/873435.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-13
下一篇2022-05-13

发表评论

登录后才能评论

评论列表(0条)

    保存