torch uint8 筛选数据

最新推荐文章于 2024-02-29 21:15:00 发布

AI算法网奇

最新推荐文章于 2024-02-29 21:15:00 发布

阅读量3.9k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/83996368

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了在PyTorch中，uint8数据类型不能进行乘法操作的特点，并讨论了如何进行数据类型转换。转换数据类型时，原始数据保持不变，主要改变的是筛选规则。此外，文章提到了torch.tensor的dtype参数，提供了八种不同的数据类型选择，其中默认类型为torch.FloatTensor。同时，device参数用于指定tensor的内存分配位置，可以是CPU或CUDA设备，并可以选择特定设备编号。

摘要由CSDN通过智能技术生成

iou > 0.5相当于uint8相当于byte类型，不能进行乘法操作。

数据类型转换：

转换后数据本身不变，只是筛选的时候规则改变了，

index=torch.Tensor([0,1,1]).type(torch.ByteTensor)
print(index)
index=index.type(torch.FloatTensor)
print(index)

import torch m=torch.Tensor([0.1,0.2,0.3]) iou=torch.Tensor([0.5,0.6,0.7]) a=iou > 0.5 index=torch.Tensor([0,1,1]).type(torch.ByteTensor) m[index]=2 m[a]=2 #a与index结果一样，因为是ByteTensor,这里0,1代表每一个位置是否有效 print(m) #tensor([0.1000, 2.0000, 2.0000]) m=torch.Tensor([0.1,0.2,0.3]) index=torch.Tensor([0,1,1]).type(torch.LongTensor)#这里0,1代表序号 #