register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[原始碼]¶
裝飾一個核心 (kernel),以便為 functional 和 (自定義) tv_tensor 類型註冊它。
有關使用細節,請參閱 如何編寫您自己的 TVTensor 類別。
使用
register_kernel
的範例
裝飾一個核心 (kernel),以便為 functional 和 (自定義) tv_tensor 類型註冊它。
有關使用細節,請參閱 如何編寫您自己的 TVTensor 類別。
使用 register_kernel
的範例