怎样使用argsort对tensor进行排序
发布于 8月前 作者 zzy380 来自其他
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) 这样的结果不对
回到顶部