Fonction torch.set_printoptions de PyTorch

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

torch.set_printoptionsIl s'agit d'une fonction de PyTorch utilisée pour définir les options d'impression des tenseurs.

Définition de fonction

torch.set_printoptions(precision, threshold, edgeitems, linewidth, profile, sci_mode)

Exemple d'utilisation

Exemple

import torch

# Définir la précision d'impression à 2 décimales
torch.set_printoptions(precision=2)

x = torch.randn(3, 3)
print(x)

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

Autres extensions