python - 如何使用 "False"- Tensorflow 中的元素将 bool 掩码的大小加倍

Question

TensorFlow 中有没有办法将带有“False”元素的 bool 掩码张量的大小加倍？换句话说，在每个元素之后我想附加一个“False”元素。

例如，我的 bool 掩码如下所示:

Mask = [True False True False]

使用“False”元素将大小加倍后，它应该如下所示:

Mask = [True False False False True False False False]

Answer 1

创建一个具有相同形状的False蒙版，将其与原始蒙版堆叠，然后使用reshape将其压平:

import tensorflow as tf
tf.InteractiveSession()

mask = tf.constant([True, False, True, False])

mask_false = tf.constant(False, shape=mask.shape)
mask_stack = tf.stack((mask, mask_false), axis=1)
tf.reshape(mask_stack, (-1,)).eval()

# array([ True, False, False, False,  True, False, False, False], dtype=bool)