目录

TVTensors

TVTensor 是torch.Tensorv2 转换在后台使用的子类将其输入分派给适当的 较低级别的内核。大多数用户不需要直接作 TVTensors。

请参阅 转换 v2 入门 TVTensors 简介,或 TVTensors 常见问题解答,了解更高级的内容 信息。

Image(data, *[, dtype, device, requires_grad])

torch.Tensor形状为 的图像的子类 。[..., C, H, W]

Video(data, *[, dtype, device, requires_grad])

torch.Tensor形状为 的视频的子类 。[..., T, C, H, W]

BoundingBoxFormat(值)

边界框的坐标格式。

BoundingBoxes(数据、*、格式、canvas_size)

torch.Tensorshape 为 的边界框的子类。[N, 4]

Mask(data, *[, dtype, device, requires_grad])

torch.Tensor用于分割和检测 Mask 的子类 shape 为 。[..., H, W]

TVTensor

所有 TVTensor 的基类。

set_return_type(return_type)

将 torch作的返回类型设置为TVTensor.

wrap(wrappee, *, like, **kwargs)

转换torch.Tensor (wrappee) 转换为相同的TVTensor子类指定为 .like

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源