PyTorch torch.set_printoptions función

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.set_printoptionsEs una función en PyTorch que se utiliza para configurar las opciones de impresión de tensores.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Configurar la precisión de impresión a 2 decimales
torch.set_printoptions(precision=2)

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones