pytorch 记录

item()把张量中的元素变成数字,比直接用索引精度要高一些

torch.unsqueeze()可以拓展一维 squeeze可以去掉一维(这两者只对维数为1的轴生效)

torch.expand 可以对张量进行扩张,但注意,只能扩张维数为1或0的轴,如下图,对于维数为2的轴进行扩张就报错了

torch.transpose()和torch.permute()可以对轴进行变换,transpose是调换,permute是按照新的顺序重构。

torch.cat(inputs, dimension=0) :在给定轴上对输入的张量进行拼接

torch.stack():沿着一个新维度对输入张量序列进行拼接

张量的索引

掩码取值


其中ge是取>的意思,

posted @   姬桑day1  阅读(21)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律
点击右上角即可分享
微信分享提示