(shapes (2, 1), (3,) mismatch) x: array([[0.666667], [1.333333]]) y: array([0.5, 1.5, 1. ]) def mean(X): D, N = X.shape mean = np.zeros((D,1)) mean = np.mean(X, axis=1, keepdims=True) return mean