TensorFlow 2.0 reduce_mean method brings empty shape tensor

Question

I\'m new with TensorFlow and I have seen this code implemented in version 2.x using compatibility methods like .compat.v1. Its working fine and brings