networks.py 文件源码

def rms_loss(self, inputs, **kwargs):
        """
        Root-mean-squared difference between <inputs> and encoded-decoded output.
        """
        loss = tf.squared_difference(inputs, self.recode(inputs, **kwargs))
        return tf.reduce_mean(
                   tf.reduce_mean(loss, axis=range(1, self.input_dims)) ** .5)