目录

数据点

数据点是张量子类,v2 v2转换在幕后使用它们来将输入分派到适当的低级内核。大多数用户不需要直接操作数据点,可以简单地依赖于数据集包装 - 例如,参见Transforms v2: 端到端目标检测示例

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

[BETA] torch.Tensor 的图像子类。

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

[BETA] torch.Tensor 视频子类。

BoundingBoxFormat(值)

[BETA] 边界框的坐标格式。

BoundingBox(data, *, format, spatial_size[, ...])

[BETA] torch.Tensor 的子类用于边界框。

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

[BETA] torch.Tensor 子类用于分割和检测掩码。

文档

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

查看文档

教程

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

查看教程

资源

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

查看资源