netprod does not support GPU
조회 수: 2 (최근 30일)
이전 댓글 표시
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
댓글 수: 0
답변 (0개)
참고 항목
카테고리
Help Center 및 File Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!