tfp.stats.histogram weights argument bug when using multiple axes
If you read the tensorflow documentation for tfp.stats.histogram(https://www.tensorflow.org/probability/api_docs/python/tfp/stats/histogram), upon reading the description of 'axis', it says : "Optional 0-D or 1-D integer Tensor with constant values." which means that tensorflow expects the axis to be passed as a tensor to generate the required results.

You can try passing the axis as a tensor, something like this : axis = tf.constant(0, 1)
 @joaozenobio,
Looks like this issue is more related to tensorflow propability repo. Could you please raise the issue in the respective repo from here for further assistance. 
https://github.com/tensorflow/probability/issues
Thank you!
 @ayushsh314, I've tried using this code now:
```
import tensorflow_probability as tfp
import tensorflow as tf
x = tf.constant([[1, 2, 3], [1, 2, 3.]])
w = tf.constant([[1, 1, 1], [1, 1, 1.]])
e = tf.constant([0, 2.5, 10])
axis = tf.constant([0, 1])
tfp.stats.histogram(x=x, edges=e, axis=axis, weights=w)
```

It gives me the same error:

```
Traceback (most recent call last):
  File "/opt/.pycharm_helpers/pydev/pydevconsole.py", line 364, in runcode
    coro = func()
  File "<input>", line 1, in <module>
  File "/usr/local/lib/python3.8/dist-packages/tensorflow_probability/python/stats/quantiles.py", line 393, in histogram
    raise ValueError('Number of dimensions of `x` and `weights` must '
ValueError: Number of dimensions of `x` and `weights` must coincide. Found: x has 2, weights has 1
```
 @tilakrayal, I will raise a new issue there and make a reference to this one, thanks. @joaozenobio,
Please feel free to move this issue to closed status, as it has been tracked in the respective repo. Thank you! Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60287">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60287">No</a>

<details><summary>Click to expand!</summary> 
 
 ### Issue Type

Bug

### Have you reproduced the bug with TF nightly?

No

### Source

binary

### Tensorflow Version

2.9.1

### Custom Code

No

### OS Platform and Distribution

Linux Ubuntu 22.04

### Mobile device

_No response_

### Python version

3.8.10

### Bazel version

_No response_

### GCC/Compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current Behaviour?

```shell
Tensorflow throws a ValueError when using weights and multiple axes at the same time, saying that the input and weights arguments should have the same shape, even though they already have.

Hint: Does not happen when using only one axis.
```


### Standalone code to reproduce the issue

```shell
import tensorflow_probability as tfp
import tensorflow as tf
x = tf.constant([[1, 2, 3], [1, 2, 3.]])
w = tf.constant([[1, 1, 1], [1, 1, 1.]])
e = tf.constant([0, 2.5, 10])
axis = (0, 1)
tfp.stats.histogram(x=x, edges=e, axis=axis, weights=w)
```


### Relevant log output

```shell
Traceback (most recent call last):
  File "/opt/.pycharm_helpers/pydev/pydevconsole.py", line 364, in runcode
    coro = func()
  File "<input>", line 1, in <module>
  File "/usr/local/lib/python3.8/dist-packages/tensorflow_probability/python/stats/quantiles.py", line 393, in histogram
    raise ValueError('Number of dimensions of `x` and `weights` must '
ValueError: Number of dimensions of `x` and `weights` must coincide. Found: x has 2, weights has 1
```
</details>