Open amirrazmjoo opened 6 months ago
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.
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.