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