register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[源码]¶
装饰一个内核，将其注册到一个函数和一个（自定义）tv_tensor 类型。
有关用法详情，请参阅 如何编写自己的 TVTensor 类。
register_kernel的示例
- 文档 >
- 转换图像、视频、框等 >
- register_kernel
-
快捷方式
