Fix NotImplementedError

✅ Solution

The "NotImplementedError" in PyTorch usually arises when a function or operation is called but the underlying implementation for a specific data type, device, or combination of features is missing. To fix this, either implement the missing functionality for the affected type/device, or, if that's not feasible, raise a `NotImplementedError` earlier with a more informative message to guide users away from unsupported usage. Consider dispatching to a suitable implementation based on the input types and devices to avoid the error.