Shortcuts

TVTensors

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

请参考 transforms v2 入门指南 了解 TVTensors 的介绍,或 TVTensors 常见问题解答 获取更多高级 信息。

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

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

视频(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 子类。