[GAN] #12 Optimize the Use of Device: Create a Helper Function for Device Management

[veeradi34]

Description: The code currently checks for the device and sends tensors to the device multiple times. This can be improved by creating a helper function.

Tasks:

Create a function to_device(tensor, device) that sends a tensor to the specified device.
Update the training loop and any other relevant sections to use this function.