gpt4 book ai didi

python - 如何在 TensorFlow 中进行 Argsort?

转载 作者:太空狗 更新时间:2023-10-29 21:36:18 26 4
gpt4 key购买 nike

如何沿第二轴对 25 x 5 x 5 矩阵(张量)进行 argsort?本质上,我正在寻找与 numpy 的 argsort 等效的 tensorflow(函数或方法),例如np.argsort(矩阵, 2)

最佳答案

在您的情况下,您可能会使用返回最高 k 值的 top_kk 可以是一维向量,定义每个维度“顶部”的值数。在你的情况下,如果你想要第二个轴集 k=[0, 5, 0] 可能会这样做。

tf.nn.top_k(matrix, k=[0,5,0], sorted=True)

不过我没有运行它。希望这有帮助

关于python - 如何在 TensorFlow 中进行 Argsort?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41985060/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com