Update torchvision/transforms/v2/functional/_utils.py

Co-authored-by: Nicolas Hug <[email protected]>

Author: pmeier

torchvision/transforms/v2/functional/_utils.py

@@ -10,6 +10,7 @@ def is_simple_tensor(inpt: Any) -> bool:
     return isinstance(inpt, torch.Tensor) and not isinstance(inpt, datapoints.Datapoint)
 
 
+# {dispatcher: {input_type: type_specific_kernel}}
 _KERNEL_REGISTRY: Dict[Callable, Dict[Type, Callable]] = {}