register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[源码]¶
修饰一个核函数,将其注册到一个函数和一个(自定义)tv_tensor 类型。
请参阅 如何编写您自己的 TVTensor 类 以了解用法详情。
使用
register_kernel
的示例
修饰一个核函数,将其注册到一个函数和一个(自定义)tv_tensor 类型。
请参阅 如何编写您自己的 TVTensor 类 以了解用法详情。
使用 register_kernel
的示例