2024-03-29 11:46:03 发布
网友
我试图对张量进行排序,并返回新排序的张量和旧张量中元素的索引。我的张量if shape(1,6,5000),我想根据最后一个轴对它进行排序,所以我得出以下结论:
(1,6,5000)
indices = tf.argsort(x, axis=-1) # collect indices # create new tensor? x[indices] # does not work
返回的索引是(1,6,5000)形的,我不知道如何创建新的张量
目前没有回答
目前没有回答
相关问题 更多 >
编程相关推荐