tf.argsort и np.argsort дали разные результаты

Question

Итак, я хочу получить argmax y0, и я проверил его в numpy и tenorflow 2, результаты которых разные. Не могу понять, почему.

maxy0 = np.amax(y0)
e0 = np.exp(y0 - maxy0)
p0 = e0 / np.sum(e0)
y0 = np.log(1e-20 + p0)
print(y0)
[[-46.0517 -46.0517 -46.0517 ... -46.0517 -46.0517 -46.0517]]

np.argsort(-y0)
array([[   11,     0, 26664, ..., 13337, 13330, 40001]])

tf.argsort(-y0)
array([[   11,     0,     1, ..., 39999, 40000, 40001]], dtype=int32)

почему они разные?

Answer 1

Попробуйте изменить стабильный аргумент на True , поскольку по умолчанию он равен False в tf.argsort. Пожалуйста, обратитесь https://www.tensorflow.org/api_docs/python/tf/argsort