返回矩阵张量中前k个最(大)值
pytorch:topk,同时返回索引和值。参考:
https://blog.csdn.net/qq_34914551/article/details/103738160
numpy:没有直接获取前k个最值的函数。而是先全部排序,再索引前k个。参考:
https://blog.csdn.net/dlhlSC/article/details/88072268
https://blog.csdn.net/SoftPoeter/article/details/86629329