diff --git a/tensorlayer/cost.py b/tensorlayer/cost.py index 662013653..f04133c17 100644 --- a/tensorlayer/cost.py +++ b/tensorlayer/cost.py @@ -72,8 +72,8 @@ def binary_cross_entropy(output, target, epsilon=1e-8, name='bce_loss'): # output = ops.convert_to_tensor(output, name="preds") # target = ops.convert_to_tensor(targets, name="target") with tf.name_scope(name): - return tf.reduce_mean(-(target * tf.log(output + epsilon) + - (1. - target) * tf.log(1. - output + epsilon))) + return tf.reduce_mean(tf.reduce_sum(-(target * tf.log(output + epsilon) + + (1. - target) * tf.log(1. - output + epsilon)), axis=1)) def mean_squared_error(output, target, is_mean=False):