search

tensorly / torch

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

Embedding layer supports CUDA Tensor #24

Closed JeremieMelo closed 2 years ago

JeremieMelo commented 2 years ago

torch version unravel_index to support CUDA tensor in Embedding layer fix minor bugs, e.g., view(-1) -> flatten(), view(out_shape) -> reshape(out_shape), warning: indices[{i}], "i" is not defined

JeanKossaifi commented 2 years ago

Awesome, thanks @JeremieMelo, merging!