快捷方式

TVTensors

TVTensors 是 torch.Tensor 的子类,v2 transforms 在内部使用它来将其输入分派到适当的底层内核。大多数用户不需要直接操作 TVTensors。

请参阅 Transforms 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(value)

边界框的坐标格式。

BoundingBoxes(data, *, format, canvas_size)

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

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

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

TVTensor

所有 TVTensors 的基类。

set_return_type(return_type)

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

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源