PyTorch torch.typename 関数
torch.typenameは、PyTorch においてデータ型の名前を取得するための関数です。テンソルまたはデータ型を入力として受け取り、その型の文字列名を返します。
関数定義
torch.typename(input)
使用例
実例
import torch
# テンソルの型名を取得する
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print(float32 テンソル型:, torch.typename(x))
# データ型の型名を取得する
dtype = torch.float64
print(float64 型:, torch.typename(dtype))
# 複素数型
y = torch.tensor([1+2j], dtype=torch.complex64)
print(complex64 テンソル型:, torch.typename(y))
# テンソルの型名を取得する
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print(float32 テンソル型:, torch.typename(x))
# データ型の型名を取得する
dtype = torch.float64
print(float64 型:, torch.typename(dtype))
# 複素数型
y = torch.tensor([1+2j], dtype=torch.complex64)
print(complex64 テンソル型:, torch.typename(y))
出力結果は:
float32 张量类型: torch.FloatTensor float64 类型: torch.DoubleTensor complex64 张量类型: torch.ComplexFloatTensor
その他の拡張
Pytorch torch リファレンスマニュアル