Check tf.multply doc

If x.shape is not the same as y.shape, they will be broadcast to a compatible shape.


  
      

      TensorFlow
  

  
    

tf.math.multiply  |  TensorFlow Core v2.5.0 2

  Returns an element-wise x * y.