register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[原始碼]¶
裝飾一個核心,將其註冊到一個函式和一個(自定義)tv_tensor 型別。
有關用法詳情,請參閱 如何編寫自己的 TVTensor 類。
register_kernel的示例
裝飾一個核心,將其註冊到一個函式和一個(自定義)tv_tensor 型別。
有關用法詳情,請參閱 如何編寫自己的 TVTensor 類。
register_kernel 的示例