import keras.backend as K
import tensorflow as tf
import numpy as np
def mean_squared_error(y_true, y_pred):
return K.mean(K.square(y_pred - y_true), axis=-1)
a = np.arange(5)
print(a)
b = np.random.randint(1,10,(5))
print(b)
c = mean_squared_error(a,b)
# 这里用下面的方式打印,即可查看teosor值
with tf.Session() as sess:
print(c.eval())