发新帖

怎样使用argsort对tensor进行排序

[复制链接]
41 0

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
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) 这样的结果不对
我知道答案 回答被采纳将会获得10 金币 + 5 金币 已有0人回答
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

2

积分
快速回复 返回顶部 返回列表