快捷方式

TVTensors

TVTensors 是 torch.Tensor 的子類,v2 transforms 在底層使用它們將輸入分派到相應的低階核心。大多數使用者無需直接操作 TVTensors。

請參閱 Transforms v2 入門,瞭解 TVTensors 的介紹,或參閱 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 子類。

BoundingBoxFormat(value)

邊界框的座標格式。

BoundingBoxes(data, *, format, canvas_size)

形狀為 [N, 4] 的邊界框的 torch.Tensor 子類。

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

形狀為 [..., H, W] 的分割和檢測掩碼的 torch.Tensor 子類。

TVTensor

所有 TVTensors 的基類。

set_return_type(return_type)

設定 TVTensor 上的 torch 操作的返回型別。

wrap(wrappee, *, like, **kwargs)

torch.Tensor (wrappee) 轉換為與 like 相同的 TVTensor 子類。

文件

獲取 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源