Use ndarray but without broadcasting