<human>:  Calculate mean across dimension in a 2D array
<bot>: Here is some code to calculate mean across dimension in a 2d array `a`

np.mean(a, axis=1)

#Warning: this code generation is experimental. Please inspect the code for any bugs before executing.