Add size field in the `array`

[kshitij12345]

Currently our array implementation assumes 1-d array. Adding a size field will allow us to store multi-dim arrays.

(We might also have to add strides optionally)

cc: @krshrimali

[kshitij12345]

@krshrimali would you like to give it a shot?

[krshrimali]

Yes, allow me one day to scope this - and I'll get back to you here. @kshitij12345

[kshitij12345]

Yes take your time. There is no hard deadline.