torch索引需要注意的小细节

发布时间 2023-08-29 09:41:52作者: 海_纳百川

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

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