快捷方式

torch.set_printoptions

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

设置打印选项。项目从 NumPy 中 shamelessly 移植。

参数
  • precision – 浮点输出的精度位数(默认 = 4)。

  • threshold – 数组元素总数,超过该数量将触发汇总而不是完整的 repr(默认 = 1000)。

  • edgeitems – 每个维度开始和结束处的数组项目数(默认 = 3)。

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

  • profile – 用于漂亮打印的合理默认值。可以使用任何上述选项进行覆盖。(defaultshortfull 中的任何一个)

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源