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