Fix RuntimeError

✅ Solution

RuntimeErrors in PyTorch often arise from mismatched data types, incorrect device placement (CPU vs. GPU), or shape incompatibilities during tensor operations. To resolve this, ensure all tensors involved in an operation have compatible dtypes and reside on the same device (e.g., using `.to(device)`) and that their shapes align appropriately through reshaping or broadcasting. Explicitly cast tensors using `.float()`, `.long()`, or similar type conversion methods when needed.