タグ

関連タグで絞り込む (2)

タグの絞り込みを解除

ChainerとCUDAに関するohnabeのブックマーク (1)

  • chainer.cuda.elementwiseを使ってGPUで処理を行う - Qiita

    はじめに Chainerのfunctionsのコードを読んでいるとcuda.elementwiseとかcuda.reduceの呼び出しを見かけることがあります。 これらは独自の処理をGPU上で実行するためのメソッドです。 Chainer functionの実装には欠かせないメソッドと言えるわけで、Chainer中級者になるために必要そうなので調査しました。 この記事ではcuda.elementwiseを扱い、cuda.reduceについては説明しません。 cuda.elementwiseの説明は以下にあります。 http://docs.chainer.org/en/stable/cupy-reference/kernel.html またSlideShareにPreferred Networksの奥田氏による解説があります。 http://www.slideshare.net/ryokuta

    chainer.cuda.elementwiseを使ってGPUで処理を行う - Qiita
  • 1