TVTensors
TVTensors 是 torch.Tensor 子类,v2 transforms 在幕后使用它们将输入分派到适当的低级内核。大多数用户不需要直接操作 TVTensors。
请参考
开始使用转换 v2 以了解 TVTensors 的介绍,或
TVTensors 常见问题解答 获取更高级的信息。
Image(数据, *[, 数据类型, 设备, 需要梯度])
|
[BETA] torch.Tensor 的图像子类。 |
Video(数据, *[, 数据类型, 设备, 需要梯度])
|
[BETA] torch.Tensor 视频子类。 |
BoundingBoxFormat(值)
|
[BETA] 边界框的坐标格式。 |
BoundingBoxes(data, *, format, canvas_size)
|
[BETA] torch.Tensor 的子类用于边界框。 |
Mask(数据, *[, 数据类型, 设备, 需要梯度])
|
[BETA] torch.Tensor 子类用于分割和检测掩码。 |
TVTensor
|
[测试版] 所有 TVTensors 的基类。 |
set_return_type(return_type)
|
[BETA] 设置 TVTensor 上的 torch 操作的返回类型。 |
wrap(组件, *, 例如, **kwargs)
|
[BETA] 将一个 torch.Tensor (wrappee) 转换为与 like 相同的 TVTensor 子类。 |