从排序索引创建同级张量

2024-03-29 11:46:03 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图对张量进行排序,并返回新排序的张量和旧张量中元素的索引。我的张量if shape(1,6,5000),我想根据最后一个轴对它进行排序,所以我得出以下结论:

indices = tf.argsort(x, axis=-1) # collect indices
# create new tensor?
x[indices] # does not work

返回的索引是(1,6,5000)形的,我不知道如何创建新的张量


Tags: 元素newif排序tfcreatenotcollect