Solve example throwing an error when executed on a GPU

Author: Nathan Cassereau

examples/backends/plot_sliced_wass_grad_flow_pytorch.py

@@ -74,7 +74,7 @@
 loss_iter = []
 
 # generator for random permutations
-gen = torch.Generator()
+gen = torch.Generator(device=device)
 gen.manual_seed(42)
 
 for i in range(nb_iter_max):
@@ -136,7 +136,7 @@ def _update_plot(i):
 loss_iter = []
 
 # generator for random permutations
-gen = torch.Generator()
+gen = torch.Generator(device=device)
 gen.manual_seed(42)
 
 alpha = 0.5