pytorch

以外にいいぞこれ。

Numpy likeなので、すんなり今までのコードに入る。