Task: Flatten Only Some Dimensions of a NumPy Array

Task: Flatten Only Some Dimensions of a NumPy Array: a task in Terminal-Lego-15k (Harbor dataset). Create a Python script that demonstrates how to flatten only some of the first dimensions of a NumPy array while keeping other dimensions intact.