Я хочу применить tf.nn.top_k к столбцам.
Предположим, у нас есть тензор p:
sim_topics = tf.constant( [[0.65 ,0.63 ,0.22 ,0.201, 0.2],
[0.3 ,0.51 ,0.1 ,0.2 , 0.4 ],
[0.2 ,0 ,0. ,0.50, 0.6 ],
[0.1 ,0 ,0.37 ,0.1 , 0.8 ]])
, если я хочу сделать это за rows, это выглядит так:
tf.cast(tf.nn.top_k(tf.nn.top_k(sim_topics,sim_topics.shape[1])[1][:,::-1], sim_topics.shape[1])[1][:,::-1],sim_topics.dtype)
Однако мне нужно сделать это над столбцами.Когда я изменяю 1 на 0 в надежде, что он даст мне результат по столбцу, вывод не имеет смысла или выдает ошибку из-за несовместимости.
Мне нужно сделать это в tf 1.12 не имеет tf.argsort