深度学习库-TFtorch
漫漫冬程
这个作者很懒,什么都没留下…
展开
-
Pytorch mask_select 函数
非常简单的函数,但是官网的介绍令人(令我)迷惑,所以稍加解释。 mask_select会将满足mask(掩码、遮罩等等,随便翻译)的指示,将满足条件的点选出来。根据掩码张量mask中的二元值,取输入张量中的指定项( mask为一个 ByteTensor),将取值返回到一个新的1D张量,张量 mask须跟input张量有相同数量的元素数目,但形状或维度不需要相同 ...原创 2018-08-14 15:21:40 · 18598 阅读 · 0 评论 -
用numpy实现topk函数(并排序)
np.argpartition 难以解决topKpytorch提供了topk函数,可以将高维数组沿某一维度(该维度共N项),选出最大(最小)的K项并排序。返回排序结果和index信息。更轻量级的numpy库提供了argpartition 和 partition,可以将最大(最小)的K项排到前K位。以argpartition为例,最小的3项排到了前3位:>>&amp原创 2019-01-24 16:37:26 · 587395 阅读 · 7 评论