From numpy to pyro (distribution shapes)

I am trying to recreate this log likelihood in pyro following the tutorial but I am not sure if this is correct/optimal:

This is the code in python using scipy:

pyro version:

The last line gives me an error(see below), any suggestions?


You might try input = input.unsqueeze(1) before the last call.