torchのnarrowメソッドがパッと見よくわからんので試してみた結果のメモ。 [Tensor] narrow(dim, index, size) ↑のサイトを参考にした。 narrowメソッドは何をするかというと、テンソル内部のある次元のある部分を切り取って返す関数である。ちなみに参照渡…

備忘録メモ。 GTX1080をUbuntu16.04,CUDA8.0でセットアップしたが，今までTITANやTeslaで動いていたコードがNaNを吐き出すようになった。使っていたライブラリはchainer。 調べたところconvolutional layerのforward計算の出力がNaNになっていることが分かっ…

はじめに chainerでforwardの計算をするときに行列の対角成分をとりたい時があったが、これを計算するfunctionsが現行のchainer(v1.19.0)に確認できなかった。 ちなみにTensorflowだとdiag_part()がこれに相当する。 http://devdocs.io/tensorflow~python/ma…