metrics.py 文件源码

def create_metric_ops(self, _inputs, labels, predictions):
        """Creates the metric op"""
        loss_mask = tf.sequence_mask(
            lengths=tf.to_int32(labels["target_len"] - 1),
            maxlen=tf.to_int32(tf.shape(predictions["losses"])[1]))
        return metrics.streaming_mean(predictions["losses"], loss_mask)