search
LoginSignup
0

posted at

[学習メモ]pytorchでよく使う関数

最短コースでわかる PyTorch &深層学習プログラミングより

view関数

numpyでいうところのreshape

item関数

スカラーに対しては、テンソルからpython本来のクラスの数値(floatまたはint)を取り出すのに使う。
1階以上のテンソルには使えない。

numpyへの変換

x_np = x.data.numpy()

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
What you can do with signing up
0