栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

tensor中数据类型的相互转换

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

tensor中数据类型的相互转换

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()
 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/864134.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号