tf.sequence_mask()

tf.sequence_mask(
    lengths,    # true的個數   （句子的真實長度）
    maxlen=None,  # 向量的總長度  （句子的最大長度）
    dtype=tf.bool,
    name=None
)

tf.sequence_mask([1, 3, 2], 5)  
# 向量的長度是5
# [[True, False, False, False, False],  從頭開始有1個true
#  [True, True, True, False, False],    從頭開始有3個true
#  [True, True, False, False, False]]   從頭開始有2個true


# 如果沒有maxlen，默認取lengths的最大值
tf.sequence_mask([[1, 3],[2,0]]) 
 # [[[True, False, False],
 #   [True, True, True]],
 #  [[True, True, False],
 #   [False, False, False]]]