This wont do the reshaping in a way that i want. the view function will reshape row wise.
here is how it will be if we do the way that you say


image.png625×681 23 KB

but i need the first subblock to be like this

[[0.803,0.6546],
[0.2339,0.9017]]