tntorch
tntorch copied to clipboard
device mismatch solved for truncated SVD
The output of truncated_svd should ideally reside in the same device as M. However, in certain cases where the tensor is low rank, the output has been observed to be on the 'cpu' device instead.