PyTorch torch.set_default_dtype 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.set_default_dtypePyTorch において、デフォルトの浮動小数点データ型を設定するための関数です。

関数定義

torch.set_default_dtype(d)

使用例

例

import torch

# デフォルトの dtype を float64 に設定
torch.set_default_dtype(torch.float64)

# テンソルを作成し、デフォルトで float64 を使用
x = torch.tensor([1.0, 2.0, 3.0])
print("デフォルト dtype:", x.dtype)

# デフォルトに戻す
torch.set_default_dtype(torch.float32)

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張