用法：

sequence_mask(lengths,maxlen=None,dtype=tf.bool,name=None )

函數參數

• lengths：整數張量，其所有值小于等于maxlen。
• maxlen：標量整數張量，返回張量的最后維度的大小；默認值是lengths中的最大值。
• dtype：結果張量的輸出類型。
• name：操作的名字。

用例子說明用法：

import tensorflow as tfa = tf.sequence_mask([1, 2, 3], 5) b = tf.sequence_mask([[1, 2], [3, 4]])with tf.Session() as sess:sess.run(tf.global_variables_initializer())print(sess.run(a))print(sess.run(b))

輸出：

[[ True False False False False]
?[ True ?True False False False]
?[ True ?True ?True False False]]

解析：maxlen是5，所以一共有5列，lengths有三個元素[1,2,3]，所以有三行，每一行分別前1、2、3個元素為True

?

[[[ True False False False]
? [ True ?True False False]]

?[[ True ?True ?True False]
? [ True ?True ?True ?True]]]

解析：因為沒有指定maxlen，故maxlen默認取lengths中的最大值4，所以一共有4列，lengths是二維數組，將其看作由兩個一維lengths組成，所以輸出也可以看作由這兩個一維lengths的輸出所組成

總結

以上是生活随笔為你收集整理的tf.sequence_mask的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。