小编Ter*_*itz的帖子

有谁知道如何提取排名2张量的每行的前n个最大值？

例如,如果我希望形状[2,4]的张量的前2个值具有值:

[[40,30,20,10],[10,20,30,40]]

所需的条件矩阵如下所示:[[True,True,False,False],[False,False,True,True]]

一旦我有了条件矩阵,我就可以使用tf.select来选择实际值.

谢谢你的帮助!