Pytorch unique

楼主: jack155861 (萧)   2021-07-04 09:32:55
各位大大好
想询问跟pytorch有关的问题
我原本有一个tensor叫做tcls_tensor,将他unique后出现了原本tensor没有的元素
我重新建立了一个tensor,内容与tcls_tensor一模一样,unique之后是正常的
想请问是什么原因吗
https://i.imgur.com/6GGar0w.jpg
楼主: jack155861 (萧)   2021-07-04 11:53:00
楼上大大您是说是因为版本问题造成原本tensor unique后产生错误结果??
作者: lycantrope (阿宽)   2021-07-04 13:26:00
我测unique也没问题,tcls_tensor确没有做其他处理吗
作者: TitanEric (泰坦)   2021-07-04 13:27:00
不过不应该有这种错误… 这看起来真的怪先放tensor到cpu上试
楼主: jack155861 (萧)   2021-07-04 18:10:00
tcls_tensor 是yolov5 predict 出来的分类结果 不知为何拿去unique 产生很诡异的东西...

Links booklink

Contact Us: admin [ a t ] ucptt.com