Desired output shape of log_prob()

[michaeldeistler]

Currently, .log_prob() returns shape N, where N is the number of theta at which log_prob had been called. Would N,1 be more desirable here?

Also, should we directly sum the returned log_probs? I.e. Evaluate the joint log_prob of a batch of samples.

See #275