Fix torch to numpy conversion when on GPU

[mantasu]

Description

A small fix to properly convert torch.Tenor into numpy.ndarray. As mentioned, converting from torch to jax does not cause the same error, at least when jax[cuda] is installed.

Fixes #1108

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)

Checklist:

• I have run the pre-commit checks with pre-commit run --all-files (see CONTRIBUTING.md instructions to set it up)
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

[pseudo-rnd-thoughts]

Thanks for the PR