Создайте маску False
той же формы, сложите ее с исходной mask
, а затем сгладьте ее с помощью 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)