tntorch icon indicating copy to clipboard operation
tntorch copied to clipboard

device mismatch solved for truncated SVD

Open amirrazmjoo opened this issue 2 months ago • 0 comments

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.

amirrazmjoo avatar Apr 10 '24 09:04 amirrazmjoo