Tensorized Matrices Error: np.unravel_index can't convert cuda device type tensor to numpy

tensorly / torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

http://tensorly.org/torch/

BSD 3-Clause "New" or "Revised" License

74 stars 19 forks source link

Closed KaidDuong closed 1 month ago

JeanKossaifi commented 5 months ago

Can you share a minimal snippet of code to reproduce the issue?

JeanKossaifi commented 1 month ago

Hi @KaidDuong - can you maybe share what the issue/fix was?