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

相关推荐
Ai思想家4 小时前
私有化部署的服务器选型与容量规划
人工智能·安全·ai
小码哥哥5 小时前
实战指南:企业网盘与AI知识库的融合架构设计与实现
人工智能
品牌测评5 小时前
AI编码平台的技术演进与行业格局——2026年生态观察
人工智能
爱查宝小二5 小时前
爱查宝 AIGC 检测与降重能力深度评测
人工智能·aigc
白拾5 小时前
【CVPR 2026】POUR:神经坍缩驱动的可证明最优表示级机器遗忘|从隐私合规与表示几何视角
人工智能·机器遗忘·表示学习·医学影像ai·cvpr 2026·pour 论文分享·神经坍缩
重庆传粉科技5 小时前
AI搜索重构互联网价值交换:GEO如何缓解全域流量流失难题
大数据·人工智能
BUG研究员_6 小时前
Runnable与LCEL
开发语言·人工智能·python
老马聊技术6 小时前
Pytorch深度学习环境配置与测试
人工智能·pytorch·python
带娃的IT创业者6 小时前
Open-Agents:当文件类型检测遇上轻量级AI代理范式
人工智能·ai代理·轻量级架构·文件类型检测·open-agents·vercel labs·内容识别