Fix transfer of data from GPU to CPU after computation#239

Open

marcovarrone wants to merge 1 commit intoMathOnco:mainfrom

marcovarrone:fix-208-gpu-compatibility

Commits on Feb 23, 2026

Add `.cpu()` calls before converting PyTorch tensors to NumPy arrays.
marcovarrone
committed