Fix TorchRuntimeError

✅ Solution

TorchRuntimeError generally arises from invalid or unsupported operations performed on tensors within a PyTorch function, often due to incorrect data types, shapes, or device placements. Inspect the specific error message and stack trace to pinpoint the problematic operation. Ensure input tensors have compatible dtypes and shapes, and that all operations are performed on the correct device (CPU or CUDA) by transferring tensors with `.to(device)` if needed.