Batch MvNormal should work. You can call .eval() on your intermediate variables to see if they are coming out with the shape you expect them to
Batch MvNormal should work. You can call .eval() on your intermediate variables to see if they are coming out with the shape you expect them to