Как я могу аргументировать матрицу 25 x 5 x 5 (тензор) вдоль 2-й оси? По сути, я ищу эквивалент (функцию или методологию) тензорного потока для argsort numpy, например. np.argsort(matrix, 2).Как Argsort в тензорном потоке?
В вашем случае вы, вероятно, можете использовать top_k, который возвращает самые высокие значения k. k может быть 1D-вектором, определяющим, сколько значений «верх» для каждого измерения. В вашем случае, если вы хотите, чтобы вторая ось, установленная k=[0, 5, 0], могла это сделать.
tf.nn.top_k(matrix, k=[0,5,0], sorted=True)
Я не запускал его. Надеюсь, что это поможет