python - 如何在一维 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？，可能性能较慢？)