register_kernel

torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)

装饰一个内核以将其注册为功能和（自定义）tv_tensor类型。

请参阅如何编写自己的TVTensor类以获取使用详情。

使用 register_kernel 的示例：

如何编写你自己的TVTensor类

How to write your own TVTensor class