目录

TVTensors

TVTensors 是 torch.Tensor 子类,v2 transforms 在幕后使用它们将输入分派到适当的低级内核。大多数用户不需要直接操作 TVTensors。

请参考 开始使用转换 v2 以了解 TVTensors 的介绍,或 TVTensors 常见问题解答 获取更高级的信息。

Image(数据, *[, 数据类型, 设备, 需要梯度])

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

Video(数据, *[, 数据类型, 设备, 需要梯度])

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

BoundingBoxFormat(值)

边界框的坐标格式。

BoundingBoxes(data, *, format, canvas_size)

torch.Tensor 的子类,用于形状为 [N, 4] 的边界框。

Mask(数据, *[, 数据类型, 设备, 需要梯度])

torch.Tensor 的子类,用于分割和检测掩码,形状为 [..., H, W]

TVTensor

所有TVTensors的基础类。

set_return_type(return_type)

设置在TVTensor上的torch操作的返回类型。

wrap(组件,  *,  例如,  **kwargs)

将一个 torch.Tensor (wrappee) 转换为与 TVTensor 子类相同的 TVTensor 子类,如同 like

文档

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

查看文档

教程

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

查看教程

资源

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

查看资源