weights_tiled = tf.tile(
      weights[:, tf.newaxis, tf.newaxis], [1, height, width])