search
LoginSignup
0

posted at

updated at

[学習メモ]テンソル生成時の注意点

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

テンソル生成時には、必ず後ろにfloat関数の呼び出しを付けてdtype(テンソル変数の要素のデータ型)を強制的にfloat32に変換するようにする。Numpy変数に対してこの処理を忘れた場合、dtypeがfloat64になり、機械学習で利用するライブラリを使うときにエラーになる

x=torch.tensor(1.0).float()

print(type(x))

print(type(x.dtype))

出力

<Class 'torch.Tensor'>
torch.float32

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