https://bestthaifoodrecipeshopppp.blogspot.com/2022/10/tensorflow-probability.html
Tensorflow Probability: InvalidArgumentError: Required Broadcastable Shapes