Is there a quick way to "sub-flatten" or flatten only some of the first dimensions in a numpy array?Is there a quick way to "sub-flatten" or flatte
Is there a quick way to "sub-flatten" or flatten only some of the first dimensions in a numpy array?Is there a quick way to "sub-flatten" or flatte