PyTorch などで Tensor の結果を numpy.testing で比較するメモ

背景

• 自前 C++ 機械学習が, PyTorch などとの結果と一致するかどうか調べたい
  • Tensor データや, 各 function(op)の計算結果(Tensor or array)など
  • データコンバーターで, 正しく weight などがコンバートできているか調べたい.

PyTorch などのレベルではお手軽に使えるものはなさそうです.

numpy にテスト関数がいくつかあります.

numpy.testing.assert_allclose

onnx-chainer での利用例が参考にし,

numpy.testing.assert_allclose を使えばよさそうです!

TODO

• 誤差があるときはテキストではわかりずらいので, matplotlib あたりで可視化できるようにする.