How to reshape array by using kotlin like Python numpy?

Question

I use tensorflow lite(kotlin android) and get output like this:[1,256,256,1], kotlin type is Array>>, thi