TVTensors
TVTensors 是 torch.Tensor 的子类，v2 transforms 在底层使用它们将输入分派到适当的低级内核。大多数用户不需要直接操作 TVTensors。
有关 TVTensors 的介绍，请参阅 开始使用 transforms v2，或有关更高级信息的 TVTensors FAQ。
Image(data, *[, dtype, device, requires_grad])
|
形状为 [..., C, H, W] 的图像的 torch.Tensor 子类。 |
Video(data, *[, dtype, device, requires_grad])
|
形状为 [..., T, C, H, W] 的视频的 torch.Tensor 子类。 |
KeyPoints(data, *, canvas_size[, dtype, ...])
|
形状为 [..., 2] 的张量的 torch.Tensor 子类，表示图像中的点。 |
BoundingBoxFormat(value)
|
边界框的坐标格式。 |
BoundingBoxes(data, *, format, canvas_size)
|
形状为 [N, K] 的边界框的 torch.Tensor 子类。 |
Mask(data, *[, dtype, device, requires_grad])
|
形状为 [..., H, W] 的分割和检测掩码的 torch.Tensor 子类。 |
TVTensor
|
所有 TVTensors 的基类。 |
set_return_type(return_type)
|
设置 TVTensor 上 PyTorch 操作的返回类型。 |
wrap(wrappee, *, like, **kwargs)
|
将 torch.Tensor（wrappee）转换为与 like 相同的 TVTensor 子类。 |
