torch索引需要注意的小细节

1.索引的数据类型必须是int,bool,有时候会出现使用float类型去索引,这个时候会报错提示数据格式不对,这个时候需要检查自己的数据类型

2.加入有个a =tensor([1,2,3]), 当a>1,会取索引tensor([1,2]) ,当a<2,会取索引tensor(0),此时的区别就是:当索引数超过一个时,会带[],只有一个时就没有[];这种区别对于根据索引取值没有影响,但是如果你使用for循环便利索引,没有[]的就会报错,所以为了保险期间,最后都对所以加上view(-1),保证数据格式都是tensor([......])

posted @   海_纳百川  阅读(25)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
本站总访问量8960309
 
点击右上角即可分享
微信分享提示