功能是产生一个mask张量,该函数的原型是:
sequence_mask(
lengths,
maxlen=None,
dtype=tf.bool,
name=None
)
其参数的含义如下:
- lengths:储存长度的张量,比如我们输入的一批文本中每个文本的真实长度,可以是任意阶
- maxlen:最大长度,比如处理文本是我们一般将每个语句的长度设置为语聊中长度最大的语句长度
输出是True or False的比lengths高一阶的张量
例
import tensorflow as tf
a=tf.sequence_mask([1, 3, 2], 5)
with tf.Session() as sess:
print(sess.run(a))