Ari*_*Ari 7 python numpy tensorflow
如何沿第二轴转动25 x 5 x 5矩阵(张量)？从本质上讲,我正在寻找与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)
我没有经营它.希望这可以帮助