The dimension of the initial state

Thank you for sharing this interesting work! I have a question regarding the shape of the initial state. In the code, the initial state has a shape of `50 x 92`, and I’m a bit confused about the second dimension. Could you please explain why the second dimension is 92 and what it represents?