我正在制作一个损失函数,我在其中使用了tf.reduce_mean(). 但它返回一个近似值。
我的代码如下:
import tensorflow as tf
real = [[1.0], [0.3]]
pred = [[0.8], [0.2]]
loss_object2 = tf.keras.losses.mean_squared_error
def loss_function(real, pred):
loss_ = loss_object2(real, pred)
print(loss_)
return tf.reduce_mean(loss_)
loss_function(real, pred)
给出以下输出:
tf.Tensor([0.04 0.01], shape=(2,), dtype=float32)
<tf.Tensor: shape=(), dtype=float32, numpy=0.024999999>
这应该只是回归0.025,为什么要回归0.024999999?
潇湘沐
相关分类