张量具有形状[?，0] --如何重塑为[?，]

Question

当src具有shape [?]时，tf.gather(src, tf.where(src != 0))返回具有shape [?, 0]的张量。我不确定一个维度的大小如何为0，我尤其不确定如何将张量改回原处。我在文档中也没有找到任何解释这一点的东西。

我尝试tf.transpose(tensor)[0]，但转置张量的第一维大小为0，无法访问！怎么了？

Answer 1

我认为你应该使用tf.not_equal对张量进行元素级的比较。

src = tf.constant([0, 1, 1, 0], dtype=tf.int8)
tf.gather(src, tf.where(tf.not_equal(src, 0))).eval(session=tf.Session())

array([[1],
       [1]], dtype=int8)

您还可以将其缩短一点，并使用tf.boolean_mask代替tf.where和tf.gather

tf.boolean_mask(src, tf.not_equal(src, 0)).eval(session=tf.Session())
array([1, 1], dtype=int8)

注意输出形状的不同。