tensorflow2.0计算两个Tensor的均方误差

tensorflow2.0计算均方误差(MSE, mean squared error)，参考代码：

import tensorflow as tf

d1=tf.random.uniform([2,3])
print(d1)
d2=tf.random.uniform([2,3])
print(d2)
loss=tf.keras.losses.mse(d1,d2)
print(loss)
print(tf.reduce_mean(loss))
# tf.Tensor(
# [[0.8498287  0.8865001  0.8501965 ]
#  [0.48766267 0.31893742 0.22491038]], shape=(2, 3), dtype=float32)
# tf.Tensor(
# [[0.36948133 0.34983194 0.7392118 ]
#  [0.42417634 0.74922156 0.9009669 ]], shape=(2, 3), dtype=float32)
# tf.Tensor([0.17702132 0.21540911], shape=(2,), dtype=float32)
# tf.Tensor(0.19621521, shape=(), dtype=float32)