怎样使用 argsort 对 tensor 进行排序

技术问答
1 
A = tf.random_uniform ((32,512,64,3))  # A.shape [32,512, 64,3]
B = A [:,:,:,0]                        # B.shape [32,512, 64]
indices = tf.argsort (B, axis=-1)      # indices.shape [32,512,64]

如何对 A 的 axis = -2 进行排序？？ A_res = tf.gather_nd (A,indices) 这样的结果不对

zzy380 2019-2-12 14:46:01

2

A_sorted = tf.gather (A,indices)

yxy_whu 发表于 2019-7-24 15:15:39

3

A_sorted = tf.gather (A,indices)

yxy_whu 发表于 2019-7-24 15:16:14