网络编程
位置:首页>> 网络编程>> Python编程>> 关于torch中tensor数据类型的转换

关于torch中tensor数据类型的转换

作者:深渊为陵  发布时间:2022-07-16 20:59:59 

标签:torch,tensor,数据类型,转换

torch中tensor数据类型转换

转换为float32类型:

points = points.type(torch.float32)

读取torch.Tensor数据类型

最近弄一下yolov5的detect.py,想提取一下检测数据,记录

关于torch中tensor数据类型的转换

a = torch.tensor(4., device='cuda:0')
print(a.item(), a.dtype, a.data, a.int())
'''
输出:
4.0
torch.float32
tensor(4., device='cuda:0')
tensor(4, device='cuda:0', dtype=torch.int32)
*********************************************
a.item()  输出里面的数字(yolo中即代表class的下标)
a.dtype输出数据类型
a.data输出当前的数据,同print(a)
a.int() 数据类型转换成整型
'''

torch.Tensor 类型,具体 data 为 tensor(1., device=‘cuda:0’)

跳转看一下类型的定义(ctrl+点击tensor)

a = torch.tensor()

可以看到定义是

def tensor(data: Any, dtype: Optional[_dtype]=None,
device: Union[_device, str, None]=None,
requires_grad: _bool=False) -> Tensor: ...

来源:https://blog.csdn.net/m0_46345373/article/details/123744198

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com