s2s.py 文件源码

def w2v_error(self):
        # mistakes = tf.not_equal(tf.argmax(self.target, 1), tf.argmax(self.encoder, 1))
        # return tf.reduce_mean(tf.cast(mistakes, tf.float32))
        y_true = tf.nn.l2_normalize(self.target, dim=-1)
        y_pred = tf.nn.l2_normalize(self.w2v_predictor, dim=-1)
        return -tf.reduce_mean(y_true * y_pred)