PyTorch torch.set_default_dtype Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.set_default_dtypeEs ist eine Funktion in PyTorch, mit der der standardmäßige Gleitkomma-Datentyp festgelegt wird.

Funktionsdefinition

torch.set_default_dtype(d)

Anwendungsbeispiel

Beispiel

import torch

# Standard-dtype auf float64 setzen
torch.set_default_dtype(torch.float64)

# Einen Tensor erstellen, der standardmäßig float64 verwendet
x = torch.tensor([1.0, 2.0, 3.0])
print("Standard-dtype:", x.dtype)

# Standard wiederherstellen
torch.set_default_dtype(torch.float32)

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen