在 Tensorflow 中为具有 N 个不同元素的长度为 L 的向量分配

Question

给定一个长度为 L 的一维张量 T，它只有 N 个不同的值，如何将其转换为长度为 L 的张量 T2，其值介于 0 和 N - 1 之间，对应于原始张量 T 的值。示例：

T  = [45, 58, 72, 33, 45, 58, 58, 33]
T2 = [ 0,  1,  2,  3,  0,  1,  1,  3]

排序并不重要，例如这也可以：

T2 = [1, 0, 2, 3, 1, 0 , 0, 3]

Answer 1

也许尝试：tf.unique(T)[1]tf.unique_with_counts(T)[1]