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博客

相关推荐
___波子 Pro Max.25 分钟前
Python参数解析默认True变False
python
逐鹿人生32 分钟前
【人工智能工程师系列】一【全面Python3.8入门+进阶】ch.3
人工智能
橙露39 分钟前
面向对象编程思想:Java 与 Python 的封装、继承与多态对比分析
java·开发语言·python
Jia ming39 分钟前
《智能法官软件项目》—法条检索模块
python·教学·案例·智能法官软件
杨浦老苏43 分钟前
本地优先的AI个人助手Moltis
人工智能·docker·ai·群晖
OBS插件网1 小时前
OBS直播如何给人脸加口罩特效?OBS口罩特效插件下载安装教程
人工智能·数码相机·语音识别·产品经理
LitchiCheng1 小时前
Mujoco 如何添加 Apriltag 并获得相机视野进行识别
人工智能·python·开源
想用offer打牌1 小时前
一站式了解Agent Skills
人工智能·后端·ai编程
一切尽在,你来1 小时前
LangGraph快速入门
人工智能·python·langchain·ai编程
啊哈哈121381 小时前
Python基本语法复盘笔记1(输入输出/字符串/列表)
开发语言·笔记·python