netprod does not support GPU
1 visualizzazione (ultimi 30 giorni)
Mostra commenti meno recenti
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 Commenti
Risposte (0)
Vedere anche
Categorie
Scopri di più su Sequence and Numeric Feature Data Workflows in Help Center e File Exchange
Prodotti
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!