helper function to handle mps and cuda

[n0w0f]

          you could also have mps ;) I typically use or write a helper function that can also handle those cases. 

(same issue with cuda device numbers)

Originally posted by @kjappelbaum in #44 (comment)