register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[源代码]¶
装饰器,用于为一个函数和一个(自定义)tv_tensor 类型注册内核。
有关用法详情,请参阅 如何编写您自己的 TVTensor 类。
使用
register_kernel
的示例
装饰器,用于为一个函数和一个(自定义)tv_tensor 类型注册内核。
有关用法详情,请参阅 如何编写您自己的 TVTensor 类。
使用 register_kernel
的示例