tf.boolean_mask(2D, 2D) 给出一维结果

Question

问题

tf.boolean_mask() 似乎相对简单：它从张量中删除不匹配条件的值。如果掩码的尺寸与目标张量相同，则条件按元素起作用。

使用ret = tf.boolean_mask(src, mask)，我发现输出尺寸与输入尺寸不匹配。

src:    Tensor("mul_3:0", shape=(?,?), dtype=int32)
mask:   Tensor("Cast_1:0", shape=(?,?), dtype=int32)
ret:    Tensor("boolean_mask/Gather:0", shape=(?,), dtype=int32)

请注意，(?,) 与 (?,?) 的行为不同，尽管我不确定为什么或如何。

---

代码

    src = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    src = tf.ones([tf.shape(src)[0], 1], tf.int32) * src

    matrix = tf.ones_like(src, tf.int32)
    matrix = tf.matrix_band_part(matrix, 3, 3) # number for mask is arbitrary
    mask = tf.cast(matrix, tf.bool)

    ret = tf.boolean_mask(tensor=src, mask=mask)

---

有什么问题？

Answer 1

见Returns section in the docs：

(N-K+1) 维张量由张量中的条目填充 对应于掩码中的 True 值。

其中N 是src 的维度，K 是mask 的维度，当N=K 时，返回总是1D，这是你的情况。