torch.set_printoptions

torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None, sci_mode=None) [исходный код]

Задать параметры вывода. Элементы беззастенчиво заимствованы из NumPy

Параметры:

• precision – Количество знаков точности для вывода чисел с плавающей точкой (по умолчанию = 4).
• threshold – Общее количество элементов массива, при превышении которого выводится сводка вместо полного repr (по умолчанию = 1000).
• edgeitems – Количество элементов массива в сводке в начале и конце каждого измерения (по умолчанию = 3).
• linewidth – Количество символов в строке для вставки разрывов строк (по умолчанию = 80). Для матриц, выводимых в сокращённом виде, этот параметр игнорируется.
• profile – Разумные настройки по умолчанию для форматированного вывода. Их можно переопределить любыми из перечисленных выше параметров. (любой из default, short, full)
• sci_mode – Включить (True) или отключить (False) научную нотацию. Если указано None (значение по умолчанию), значение определяется torch._tensor_str._Formatter. Это значение автоматически выбирается фреймворком.

Пример:

>>> # Limit the precision of elements
>>> torch.set_printoptions(precision=2)
>>> torch.tensor([1.12345])
tensor([1.12])
>>> # Limit the number of elements shown
>>> torch.set_printoptions(threshold=5)
>>> torch.arange(10)
tensor([0, 1, 2, ..., 7, 8, 9])
>>> # Restore defaults
>>> torch.set_printoptions(profile='default')
>>> torch.tensor([1.12345])
tensor([1.1235])
>>> torch.arange(10)
tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])