Question

如何沿第二轴转动25 x 5 x 5矩阵（张量）？从本质上讲，我正在寻找与numpy的argsort相当的张量流（例如函数或方法），例如： np.argsort(matrix, 2)。

Answer 1

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

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

我没经理过。希望这有帮助

Answer 2

作为参考，Tensorflow现在支持tf.argsort。

示例：

h1