TensorFlow与深度学习——张量排序
TensorFlow与深度学习——张量排序
一、 tf.sort / argsort
tf中内置sort和argsort函数,sort用于排序升序,argsort返回排序之后各个元素所在的位置信息,默认也是升序排序。下边是代码:
##列表排序
a= tf.random.shuffle(tf.range(5)) ##[0 2 3 4 1]
b = tf.sort(a,direction='DESCENDING')
#print(b)# tf.Tensor([4 3 2 1 0], shape=(5,), dt
原创
2020-12-09 17:25:53 ·
476 阅读 ·
0 评论