python - Tensorflow:如何使 tf.unique 的返回值与输入的大小相同

Question

根据 https://www.tensorflow.org/api_docs/python/tf/unique , tf.unique(x) 返回一个元组 (y, idx), y 的形状是 (?, ) 期间未知构建时间。无论如何，我可以填充 y 以匹配输入大小 x 吗？

例如，

# tensor 'x' is [1, 1, 2, 4, 4, 4, 7, 8, 8]
y, idx = unique(x)
y ==> [1, 2, 4, 7, 8]
idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]

我想让 y = [1, 2, 4, 7, 8, 0, 0, 0, 0]

Answer 1

您可以在调用tf.unique 后使用tf.pad 来填充零。

x = tf.placeholder(tf.int32, shape=(None))
y, idx = tf.unique(x)
y = tf.pad(y,[[0,(tf.shape(x) - tf.shape(y))[0]]])

sess = tf.InteractiveSession()
print(sess.run(y, {x:np.random.randint(0,10, (10), dtype=np.int32)}))