register_kernel¶
- torchvision.transforms.v2.functional.register_kernel(functional, tv_tensor_cls)[源代码]¶
装饰内核以将其注册到函数和(自定义)tv_tensor 类型。
有关用法详细信息,请参阅 如何编写自己的 TVTensor 类。
使用
register_kernel
的示例如何编写自己的 TVTensor 类
装饰内核以将其注册到函数和(自定义)tv_tensor 类型。
有关用法详细信息,请参阅 如何编写自己的 TVTensor 类。
使用 register_kernel
的示例