TopK

  • 详解torch.topk函数

    一、torch.topk函数 在深度学习领域中,我们通常需要对张量进行排序(如特征选择、模型解释等),而PyTorch中的torch.topk()函数则是我们在进行此类操作时候的一…

    编程 2024-12-26
  • TopK算法详解

    一、TopK算法例题 TopK算法是一种寻找数据中前K大或前K小的算法,在日常开发中经常会遇到需要寻找数据中前K大或前K小的情况,例如寻找销售额前十的商品或者寻找某个数据集中最热门…

    编程 2024-11-16