NN 実装のための 行列演算・画像演算 活用 全結合Affin処理の例 - Qiita

cuBLAS と NPP を使用して、GPUで動くＮＮの自前実装を行ってます。用いた関数群を紹介します。 背景 深層学習の理解を深めようと自前で実装を行ってます。 GPU速い。15K￥位のgeforce GTX 1050tiでも、CPUに比べ10～100倍以上の実行速度があることを体感しました。 GPUで演算を行わせるため、CUDA Toolkit の行列演算(cuBLAS)と 画像演算(NPP)のライブラリを活用します。 本記事では、行列演算と画像演算の活用しての、NN の全結合のAffin処理 を 公開します。 参考文献 cuBLAS CUDA ToolKit About BLAS - BLASの簡単な使い方 NPP CUDA Toolkit IPP インテル(R) インテグレート パフォーマンス・プリミティブ リファレンス・マニュアル(PDF) cuBLAS 行列の積演算 図示 ↓の