Some question about Mamba file

[rio-kiomi]

hello!
I have some questions about the Mamba file.

First, in Figure 2, when computing the hidden state for the input $${D1}$$, the first row of $$A'$$ in Figure 1 is transformed into a diagonal matrix for the calculation. This corresponds to the values in Figure 2 for $${D1}$$: $$[1.49, 0.67], [1.41, 0.71], [1.45, 0.69]$$.

However, when computing the hidden state for the inputs $${D2}$$, $${D3}$$ and $${D4}$$, I noticed what seems to be an error. Shouldn't{ $${D2}$$ } { $${T2}$$}, according to the corresponding relationship, result in the diagonal matrix $$[2, 0.71]$$ as shown in Figure 2?

Additionally, thank you for this clear explanation of model expressions!

fig.1

fig.2