Table of Contents
Shortcuts

torch.set_printoptions

torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None, sci_mode=None)[源代码]

设置打印选项。内容无耻地借鉴自NumPy

Parameters

  • 精度 – 浮点数输出的精度位数（默认值 = 4）。

  • 阈值 – 触发汇总而不是完整 repr 的数组元素总数（默认值 = 1000）。

  • edgeitems – 每个维度开始和结束时摘要中的数组项数（默认值为3）。

  • linewidth – 每行字符数，用于插入换行符（默认值 = 80）。阈值矩阵将忽略此参数。

  • 配置文件 – 用于美化打印的合理默认值。可以使用上述任何选项进行覆盖。（可以是默认、简短、完整中的任意一个）

  • sci_mode – 启用（True）或禁用（False）科学计数法。如果指定为None（默认值），则该值由torch._tensor_str._Formatter定义。该值由框架自动选择。

示例：

>>> # 限制元素的精度
>>> torch.set_printoptions(precision=2)
>>> torch.tensor([1.12345])
tensor([1.12])
>>> # 限制显示的元素数量
>>> torch.set_printoptions(threshold=5)
>>> torch.arange(10)
tensor([0, 1, 2, ..., 7, 8, 9])
>>> # 恢复默认设置
>>> 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])
优云智算