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是取>的意思,
分类:
pytorch学习
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律