捷徑

TVTensors

TVTensor 是 torch.Tensor 子類別,v2 transforms 在底層使用它們來將輸入分派到適當的底層核心。 大多數使用者不需要直接操作 TVTensor。

請參閱 transforms v2 入門 以了解 TVTensor 的介紹,或參閱 TVTensor 常見問題 以取得更進階的資訊。

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

所有 TVTensor 的基底類別。

set_return_type(return_type)

設定 TVTensor 上 torch 運算的返回類型。

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

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

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得針對初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得您的問題解答

檢視資源