发新帖

怎样使用argsort对tensor进行排序

[复制链接]
1481 2

快来加入 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 金币 已有2人回答
本楼点评(0) 收起

精彩评论2

yxy_whu  TF荚荚  发表于 2019-7-24 15:15:39 | 显示全部楼层
本帖最后由 yxy_whu 于 2019-7-24 15:33 编辑

A_sorted = tf.gather(A,indices)
本楼点评(0) 收起
yxy_whu  TF荚荚  发表于 2019-7-24 15:16:14 | 显示全部楼层
本帖最后由 yxy_whu 于 2019-7-24 15:33 编辑

A_sorted = tf.gather(A,indices)
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

2

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