Fonction torch.dtype de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.dtypeC'est un objet dans PyTorch utilisé pour représenter le type de données d'un tenseur, et non une fonction. Il est utilisé pour obtenir ou spécifier le type de données d'un tenseur.

Description du type de données

torch.float32 - 32位浮点数
torch.float64 - 64位浮点数（双精度）
torch.float16 - 16位浮点数（半精度）
torch.bfloat16 - 16位浮点数（Brain Float）
torch.complex32 - 32位复数
torch.complex64 - 64位复数
torch.complex128 - 128位复数
torch.int8 - 8位整数
torch.int16 - 16位整数
torch.int32 - 32位整数
torch.int64 - 64位整数
torch.bool - 布尔类型
torch.uint8 - 无符号8位整数

Exemples d'utilisation

Exemple

import torch

# Créer des tenseurs de différents types de données
x1 = torch.tensor([1.0, 2.0, 3.0])
x2 = torch.tensor([1, 2, 3], dtype=torch.int32)
x3 = torch.tensor([True, False, True], dtype=torch.bool)

# Obtenir le type de données d'un tenseur
print("x1 dtype:", x1.dtype)
print("x2 dtype:", x2.dtype)
print("x3 dtype:", x3.dtype)

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions