如何省去零在1-d tensorflow張量

Question

如果我有像張量：

[0 235 0 213 1006 0. 0. 0. 0. 77. 0 234 。]

如何獲得沒有零的相同矢量？

我想我的結果是：[235。 213. 1006 77 234]

Answer 1

是的，有純tensorflow功能的方法：

a = tf.Variable([0.235, 0.213, 1006.0, 0, 0, 0, 0, 77, 0,234.0]) 
session = tf.Session() 
initialiser_op = tf.global_variables_initializer() 
session.run(initialiser_op) 
a_without_zeroes = tf.boolean_mask(a, tf.not_equal(a, 0)) 

tf.py_func是另一種解決方案，使其與numpy的工作，但我期望的缺點相比純辦法，主要是你不會得到梯度計算等（在graphdef中沒有序列化，沒有分佈式，沒有gpu ?,可能性能較差？）