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 子类。 |