import tensorflow as tf
a = tf.sequence_mask([1,2,3],4)
b = tf.sequence_mask([[1,2],[3,4]])
w = tf.constant([[1.,2,3],[1,2,1],[1,2,3],[1,2,1]])
mask = tf.sequence_mask([1,3,2,2],3, dtype=tf.float32)
w_mask=w*mask
with tf.Session()as sess:
sess.run(tf.global_variables_initializer())print(sess.run(a))print(sess.run(b))print(sess.run(w))print(sess.run(mask))print(sess.run(w_mask))