Given a matrix of shape (r, c, 1) like the below:
[[[1], [4], [5]], [[1], [7], [6]], ... [[9], [2], [8]]]
I would like it to be reshaped as (r