def take_along_axis(large_array, indexes):
<EOL>if len(large_array.shape) > len(indexes.shape):<EOL><INDENT>indexes = indexes.reshape(indexes.shape + tuple([<NUM_LIT:1>] * (len(large_array.shape) - len(indexes.shape))))<EOL><DEDENT>return np.take_along_axis(large_array, indexes, axis=<NUM_LIT:0>)<EOL>
Take along axis