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)})) 

Answer 2

你需要單獨處理的問題。

y = [1, 2, 4, 7, 8] 
idx = [0, 0, 1, 2, 2, 2, 3, 4, 4] 

for ii in range(0, len(idx) - len(y)): 
    y.append(0) 

print(y) 

輸出是

[1, 2, 4, 7, 8, 0, 0, 0, 0] 

Answer 3

使用設置，找到獨特的..diff LEN和追加尾隨0

x = [1, 1, 2, 4, 4, 4, 7, 8, 8] 
length = len(x) 

print(length) 
y = list(set(x)) 

print(y) 
pad = length - len(list(set(x))) 

for index in range(0,pad): 
    y.append(0) 


print(y) 

輸出

9 
[1, 2, 4, 7, 8] 
[1, 2, 4, 7, 8, 0, 0, 0, 0]