pytorch 今日小知识1——torch.set_printoptions

torch.set_printoptions

复制代码
torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None, sci_mode=None)

这个方法是来修改pytorch中的打印选项的(其实就是和numpy中更改打印方法一样),就是使用print打印tensor时,我们显示的元素精度,最多显示元素个数等一系列的有关操作的选项

参数 precision=None,此参数的意思是显示浮点tensor中元素的精度(显示到小数点后几位),默认是4

threshold=None,由于我们的进行训练网络时,tensor都会很大,里面的数据很多,不方便全部显示出来,显示超过一定的个数(注意是个数,是行数× \times×列数,而不是行数或者列数)后就会进行折叠。此参数是指定tensor的数目超过多少时开始显示进行折叠。默认为1000

edgeitems=None此参数也与折叠有关,折叠后只显示前面的数据和后面的数据,此参数设置显示的前面与后面的数据的行数 (注意这里是行数而不是个数),默认为3,如上图折叠后头尾各显示3行,更改为5行后如下所示:

sci_mode=None这方法是来指定显示的数字是否使用科学计数法,可以选择指定True或者False,如果选择None,那么是True还是False会由torch._tensor_str._Formatter来定义。值会自动的由框架来选择。一般默认为False

参考 pytorch每日一学10(torch.set_printoptions())更改打印设置-CSDN博客

相关推荐
loulanyue_6 小时前
10000P国产算力破壁上线,算电协同与超智融合打破算力孤岛:高文院士图灵奖得主纵论世界算力网新纪元——读2026世界算力网大会实录
人工智能·大模型·架构设计·高性能计算·算力网络
Thneonl6 小时前
0.18 也被 DENY:治理必须是台确定性机器
人工智能·架构
小嘉丶学长6 小时前
Day08 · 字符串(二)
服务器·开发语言·python·学习·1024程序员节
老猫聊AI6 小时前
Qwen-image-2.1 本地部署保姆级教程 | 首页
人工智能·llm
FridaySunny6 小时前
Agent 工作交接:继续、压缩还是新开?120 次运行的会话策略实验
人工智能
小毕超6 小时前
从推理到训练:Intel Ultra 核显 GPU 加速 YOLO 模型训练
pytorch·yolo·gpu·intel
BD_Marathon6 小时前
【Python】列表推导式
python
狂潮lister6 小时前
狂潮 · 元理论v1.0-原创架构设计-AI 网关 | 第9篇:KC-AI‑Gateway(一套可审计的大模型网关完整架构实践)|【完整版】
人工智能·后端
林澈在路上6 小时前
AI做歌工具怎么拆人声伴奏?分轨导出指南
人工智能·ai音乐·melo音乐
方方洛6 小时前
ray教程-14-RayTune与超参搜索
人工智能·分布式·机器学习