Pytorch控制打印矩阵的格式

25 篇文章 7 订阅

原文地址

分类目录——Pytorch

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

直接用程序来说明

  • 生成测试数据

    import torch
    
    torch.random.manual_seed(0)	# 固定每次生成的数据相同
    
    tensor = torch.rand(100, 9)-0.5
    print(tensor)	# 在默认的显示设置下进行print
    

    效果(部分)如下图

    1585033499309

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

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

    torch.set_printoptions(
        precision=2,    # 精度,保留小数点后几位,默认4
        threshold=1000,
        edgeitems=3,
        linewidth=150,  # 每行最多显示的字符数,默认80,超过则换行显示
        profile=None,
        sci_mode=False  # 用科学技术法显示数据,默认True
    )
    print(tensor)
    

    其中

    • precision=2

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

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

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

    • linewidth=150,

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

    • profile=None

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

      	# if profile == "default":
          #     PRINT_OPTS.precision = 4
          #     PRINT_OPTS.threshold = 1000
          #     PRINT_OPTS.edgeitems = 3
          #     PRINT_OPTS.linewidth = 80
          # elif profile == "short":
          #     PRINT_OPTS.precision = 2
          #     PRINT_OPTS.threshold = 1000
          #     PRINT_OPTS.edgeitems = 2
          #     PRINT_OPTS.linewidth = 80
          # elif profile == "full":
          #     PRINT_OPTS.precision = 4
          #     PRINT_OPTS.threshold = inf
          #     PRINT_OPTS.edgeitems = 3
          #     PRINT_OPTS.linewidth = 80
      
    • sci_mode=False

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

    如上设定之后显示如下

    1585033983750

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

  • 相关文献

    Numpy控制多维矩阵的显示格式

  • 7
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

BBJG_001

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值