PyTorch torch.typename 関数


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

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))

出力結果は:

float32 张量类型: torch.FloatTensor
float64 类型: torch.DoubleTensor
complex64 张量类型: torch.ComplexFloatTensor

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

その他の拡張