Is this correct @jabooth? I thought we said we wanted this to be: `n_points x n_dims xn_params`