InvalidArgumentError (see above for traceback): tags and values not the same shape
I'm using Tensorflow 1.13 and trying to create a custom metric for the segmentation task and to show on tensorboard, but I get a weird error. Here is my code
# ...
value = tf.reduce_mean(pred * ground_truth, axis=(1, 2))
tf.summary.scalar('loss value', value)
InvalidArgumentError (see above for traceback): tags and values not the same shape