fix bug in dims attributes of states returned by stochastic solver

Fixing bug reported in

https://groups.google.com/forum/#!topic/qutip/JPgZOp076jM