- 链接地址:https://guotong1988.blog.csdn.net/article/details/122881121
- 链接标题:从 [batch_size, seq_length] 的 mask sequence 构建 mask 矩阵_mask: batch size, 1, sequence length (bool)-CSDN博客
- 所属网站:guotong1988.blog.csdn.net
- 被收藏次数:9927
文章浏览阅读598次。# input_mask = [batch_size, seq_length] input_shape = get_shape_list(input_mask, expected_rank=2) seq_length = input_shape[1] to_mask = tf.cast( tf.reshape(to_mask, [batch_size, 1, seq_length]), tf.float32) # broadcast_ones = [batch_size.._mask: batch size, 1, sequence length (bool)
版权声明:本文发布于特牛网址导航 内容均来源于互联网 如有侵权联系删除