Investigate creating tensors directly on GPU

[ElliottKasoar]

Currently, data is (probably) explicitly moved to the target device, rather than created directly on the desired device, as is preferable.

Creating the tensor directly on the device would be closer to the previous form of the code (see changes for GPUs), although that exact implementation did not seem to work.