PyTorch torch.get_default_dtype 関数

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

torch.get_default_dtypePyTorch で現在のデフォルト浮動小数点データ型を取得するための関数です。

関数定義

torch.get_default_dtype()

使用例

例

import torch

# デフォルトのdtypeを取得
dtype = torch.get_default_dtype()
print("デフォルト dtype:", dtype)

出力結果は次のとおりです。

默认 dtype: torch.float32

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

その他の拡張