如何在坦索弗洛的阿格斯特？

Question

我如何在第二轴上讨论一个25x5x5矩阵(张量)？本质上，我正在寻找tensorflow的等价(功能或方法)与numpy的讨论，例如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。

示例

import tensorflow as tf


tensor = tf.constant(
    [
        [8, 7, 11],
        [5, 3, 4],
        [17, 33, 23],
    ]
)

arg_sort_op = tf.argsort(tensor, axis=-1)

with tf.Session() as sess:
    out = sess.run(arg_sort_op)

print(out)

#  [[1 0 2]
#   [1 2 0]
#   [0 2 1]]