James Hensman
James Hensman
Actually, it seems to be even simpler than that. here's an imprved MWE: ```python def foo(d, i): return d[i] ``` raises same exception!
@dustinvtran is there a neat way to make the kl-divergences defined in tfd broadcast nicely? It's annoying that the `tile` op is killing us.
You're right Akash. There's some more code coming: watch this space!
Hi @FrankD Yes, this is a bug! Let me take a look.