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