tensorly / torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
http://tensorly.org/torch/
BSD 3-Clause "New" or "Revised" License
70 stars 18 forks source link

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

Open KaidDuong opened 4 months ago

JeanKossaifi commented 4 weeks ago

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