loss = tf.reduce_mean(keras.losses.mean_squared_error)

来源:3-12 tf.GradientTape与tf.keras结合使用

王浩同学

2021-05-16

keras.losses.mean_squared_error(y_batch,y_pred) 不是直接计算均方差了吗? 为什么外面加上reduce_mean

写回答

1回答

正十七

2021-05-20

我们在训练的时候是每次输入多个样本,mean_squared_error会把每个样本的loss值计算出来,所以我们需要在多个样本上求均值。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程