Fix NotImplementedError

✅ Solution

NotImplementedError in PyTorch usually arises when a called function or operation lacks a specific implementation for the given input, often related to hardware (like CUDA) or data type. To resolve it, either implement the missing functionality for the specific data type/device combination or use a different function or data type that has a complete implementation, potentially casting tensors to supported types beforehand. If using custom operators, ensure they are correctly defined and registered for the target device (CPU or CUDA).