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