Diagnostics(...).summary() failing on Hierarching Modeling tutorial

[feynmanliang]

Repro: #954 (comment)

Expected: Calling summary() returns a dataframe of summary statistics

Actual:

RuntimeError: view size is not compatible with input tensor's size and stride
(at least one dimension spans across two contiguous subspaces). Use
.reshape(...) instead.