netprod does not support GPU
1 vue (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Hi, I have implemented an LSTM neural network using matlab. LSTM includes gates and i used netprod to implement them. It is running fine (but slow...) on the CPU.
I am trying to make it run on the GPU but netprod is not supported for GPU computations. I have implemented the LSTM using this blog post. below is what i have implemented.
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/159979/image.png)
Below is a crop of my network with red boxes showing the problematic parts:
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/159982/image.png)
Is there a way around it? I need something like netprod that runs on the GPU.
Thanks, Oshri
0 commentaires
Réponses (0)
Voir également
Catégories
En savoir plus sur Sequence and Numeric Feature Data Workflows dans Help Center et File Exchange
Produits
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!