conjugate transpose crashes model

[Chris]

Hi all,

I am in the process of building a quantum model and  discovered that conjugate transpose crashes the model. In the google colab notebook below, I provide a simple example. Here is what I found:

1. The model runs in all cases if `conjugate = False` in `tf.transpose`

2. The model runs if `conjugate = True` and only one chain is ran

3. The model does not run if `conjugate = True` and multiple chains are ran

Any help would be greatly appreciated.

Example:

https://colab.research.google.com/drive/1hYFFrXB1YzOz65WHYHL_aZOqxbaB5T_x?usp=sharing