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

相关推荐
逻辑君4 分钟前
科技逆向外语|20260808
人工智能·科技·机器学习
Dawson Zhu6 分钟前
面向AI自动化分析的数据仓库建模实践
人工智能·语言模型·架构·制造
民乐团扒谱机8 分钟前
【微实验】谐波乘积谱(HPS)算法深度解析:原理、数学与代码实现
开发语言·人工智能·python·算法·语音识别·音乐
tech讯息13 分钟前
Agentic BI 云方案选型:哪些平台可支持业务人员自然语言查数并分析原因?
人工智能
黄焖鸡能干四碗15 分钟前
信息安全保障方案(Word文件)
大数据·网络·人工智能·架构·区块链
致Great16 分钟前
Anthropic 终于把 Claude Code 怎么省 Token 讲明白了。
人工智能
暗黑小白19 分钟前
参数从哪来、何时来 —— 提取时机与平台化 Slot 管理
人工智能·后端·大模型·ai agent
ShallWeL28 分钟前
【机器学习】(40)—— 流水线监控
人工智能·机器学习·流水线监控
小白勇闯网安圈36 分钟前
Django Ajax、批量操作与分页实践
python·django
小易测试笔记43 分钟前
芯片ESD测试仪是什么?主要有哪几种不同类型?
人工智能