如何沿第二轴对 25 x 5 x 5 矩阵（张量）进行 argsort？本质上，我正在寻找 numpy 的 argsort 的 TensorFlow 等效项（函数或方法），例如np.argsort(matrix, 2).
在你的情况下你可能会使用top_k返回最高的k价值观。k可以是一个一维向量，定义每个维度“顶部”有多少个值。在你的情况下，如果你想要设置第二个轴k=[0, 5, 0]可能会做到。
tf.nn.top_k(matrix, k=[0,5,0], sorted=True)
我没有运行它。
希望这可以帮助
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
