Pytorch控制打印矩阵的格式

诸如长序列单行显示,全部显示(不缩略显示),精度(保留小数点后几位),是否科学计数法显示等等。

直接用程序来说明

  • 生成测试数据

    效果(部分)如下图

    1585033499309

    可以看到默认设定,过长的会换行,默认保留小数点后4位,默认进行科学计数法显示

  • 通过torch.set_printoptions()控制显示格式

    其中

    • precision=2

    精度,保留小数点后几位,默认4

    • threshold=1000
      最多可现实的Array元素个数,默认1000;
      限制的是基本元素个数,如3*5的矩阵,限制的是15而非3(行);
      如果超过就采用缩略显示;
      设置为inf全部显示

    • edgeitems=3
      在缩略显示时在起始和默认显示的元素个数(对多个维度同时有效)

    • linewidth=150,

    每行最多显示的字符数,默认80,超过则换行显示

    • profile=None

    3种预定义的显示模板,可选’default’、’short’、’full’

    • sci_mode=False

    用科学技术法显示数据,默认True

    如上设定之后显示如下

    1585033983750

    应该注意这只是设置print出来的显示格式,并不会影响到数据在内存中运算的精度

You may also like...

发表评论

电子邮件地址不会被公开。 必填项已用*标注