最短コースでわかる 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