Python tf.sort用法及代碼示例

對張量進行排序。

用法

tf.sort(
    values, axis=-1, direction='ASCENDING', name=None
)

用法：

a = [1, 10, 26.9, 2.8, 166.32, 62.3]
tf.sort(a).numpy()
array([  1.  ,   2.8 ,  10.  ,  26.9 ,  62.3 , 166.32], dtype=float32)

tf.sort(a, direction='DESCENDING').numpy()
array([166.32,  62.3 ,  26.9 ,  10.  ,   2.8 ,   1.  ], dtype=float32)

對於多維輸入，您可以控製沿哪個軸應用排序。默認的axis=-1 對最裏麵的軸進行排序。

mat = [[3,2,1],
       [2,1,3],
       [1,3,2]]
tf.sort(mat, axis=-1).numpy()
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]], dtype=int32)
tf.sort(mat, axis=0).numpy()
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]], dtype=int32)

也可以看看：

• tf.argsort ：類似於排序，但它返回排序索引。
• tf.math.top_k ：返回固定數量的頂部值和相應索引的部分排序。