How do you flatten the last two dimensions of an numpy array?

Question

For example, given a numpy array of dimensions (132, 82, 100), the resultant dimensions would be (132, 8200)

Just create a new array with the dimensions you want and insert the values as you would like them. — Almog-at-Nailo, Jun 13 '21 at 12:32
Does this answer your question? [How to flatten only some dimensions of a numpy array](https://stackoverflow.com/questions/18757742/how-to-flatten-only-some-dimensions-of-a-numpy-array) — yann ziselman, Jun 13 '21 at 12:34

score 1 · Accepted Answer · edited Jun 13 '21 at 12:38

1

You can use reshape() function to change shape of the array. Using -1 as a parameter to reshape tells numpy to infer the dimension there.

arr = np.zeros((132, 82, 100))
arr = arr.reshape(*arr.shape[:-2], -1)
print(arr.shape)
# Out: (132, 8200)

edited Jun 13 '21 at 12:38

yann ziselman

answered Jun 13 '21 at 12:33

Shubham

1 Answers1