Effacer les filtres
Effacer les filtres

netprod does not support GPU

1 vue (au cours des 30 derniers jours)
oshri
oshri le 20 Jan 2017
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.
Below is a crop of my network with red boxes showing the problematic parts:
Is there a way around it? I need something like netprod that runs on the GPU.
Thanks, Oshri

Réponses (0)

Catégories

En savoir plus sur Sequence and Numeric Feature Data Workflows dans Help Center et File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by