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])