特牛网址导航

从 [batch_size, seq_length] 的 mask sequence 构建 mask 矩阵_mask: batch size, 1, sequence length (bool)-CSDN博客

网友收藏
文章浏览阅读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)