netprod does not support GPU
1 Ansicht (letzte 30 Tage)
Ältere Kommentare anzeigen
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 Kommentare
Antworten (0)
Siehe auch
Kategorien
Mehr zu Sequence and Numeric Feature Data Workflows finden Sie in Help Center und File Exchange
Produkte
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!