Fix device mismatch when using non-default GPUs.

[imxtx]

The cuda function will create the tensor on a default device that may differ from the device of tensor 'inverse_transform'.