register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[source]¶
[BETA] Decorate a kernel to register it for a functional and a (custom) tv_tensor type.
See How to write your own TVTensor class for usage details.
Examples using
register_kernel
:How to write your own TVTensor class
How to write your own TVTensor class