Open
Description
If I undestand it correctly than currently we don't specify anywhere if we want a tensor on the CPU or GPU.
Would it make sense that the MRData class has a function .cuda() and .cpu()? This would then e.g. move _data and _mask to GPU/CPU for ImageData and _t1 _rho and _mask for QMRIData.
How do you normally specify the device? Is this a global variable you set and then everything is put to the same device or do you specify this for each tensor?
Metadata
Metadata
Assignees
Labels
No labels