oneMKL的生成随高斯分布的随机数


前言

本文主要介绍oneMKl生成随高斯分布的随机数的简要使用方法,在看本文之前请先安装oneMKL,具体步骤请查看oneMKl安装步骤
官方参考文档
英特尔® oneAPI 数学核心函数库 (oneMKL) 提供了一组例程,用于实现常用的伪随机、准随机或非确定性随机数生成器,这些生成器具有连续和离散分布。为了提高性能,所有这些例程都是使用对高度优化的基本随机数生成器 (BRNG) 和矢量数学函数(VM)的调用开发的。

主要步骤

oneMKL随机数生成器包含在mkl.h库中,因此可以直接导入调用生成,使用步骤如下所示:

#include <mkl.h>

// 1、创建或初始化随机数流(控制台)
VSLStreamStatePtr stream;
auto status = vslNewStream( &stream, brng, seed );
// 2、调用oneMKL随机数算法
status = vsRngGaussian( method, stream, n, r, a, sigma );
// 3、删除随机数流(控制台)
status = vslDeleteStream(&stream);

一、创建或初始化随机数流(控制台)

1、核心代码

#include <mkl.h>

VSLStreamStatePtr stream;
auto status = vslNewStream( &stream, brng, seed );

2、算法描述

这个函数的目标是创建一个新的随机数流,并用一个随机数种子初始化它

3、输入参数

名字类型描述
streamVSLStreamStatePtr*流状态描述符
brngconst MKL_INT用于初始化流的基本生成器的索引,该值使用的是oneMKL定义的常量,参见如下brng 参数的定义
seedconst MKL_UINT随机数种子

4、输出参数

该函数运行完后的所有设置都保存在stream中,可以通过其返回的值status进行判断,判断如下:

名字描述
VSL_ERROR_OK指示无错误,执行成功
VSL_STATUS_OK指示无错误,执行成功
VSL_RNG_ERROR_INVALID_BRNG_INDEXbrng索引无效
VSL_ERROR_MEM_FAILURE系统无法为流分配内存
VSL_RNG_ERROR_NONDETERMINISTIC_NOT_SUPPORTED不支持非确定性随机数生成器
VSL_RNG_ERROR_ARS5_NOT_SUPPORTED运行应用程序的 CPU 不支持 ARS-5 随机数生成器

5、brng 参数的定义

名字描述
VSL_BRNG_MCG3131 位乘法全等生成器
VSL_BRNG_R250广义反馈移位寄存器发生器
VSL_BRNG_MRG32K3A具有两个 3 阶分量的组合多重递归生成器
VSL_BRNG_MCG5959 位乘法全等生成器
VSL_BRNG_WH一组 273 个 Wichmann-Hill 组合乘法同余生成器
VSL_BRNG_MT19937Mersenne Twister 伪随机数生成器
VSL_BRNG_MT2203一组 6024 Mersenne Twister 伪随机数生成器
VSL_BRNG_SFMT19937面向 SIMD 的快速梅森费尔托斯特伪随机数生成器
VSL_BRNG_SOBOL基于32位格雷码的生成器,为尺寸生成低差异序列;还提供用户定义的尺寸,1 ≤ s ≤ 40
VSL_BRNG_NIEDERR基于32位格雷码的生成器,为尺寸生成低差异序列;还提供用户定义的尺寸,1 ≤ s ≤ 318
VSL_BRNG_IABSTRACT整数数组的抽象随机数生成器
VSL_BRNG_DABSTRACT用于双精度浮点数组的抽象随机数生成器
VSL_BRNG_SABSTRACT用于单精度浮点数组的抽象随机数生成器
VSL_BRNG_NONDETERM非确定性随机数生成器
VSL_BRNG_PHILOX4X32X10基于 Philox4x32-10 计数器的伪随机数生成器
VSL_BRNG_ARS5基于 ARS-5 计数器的伪随机数生成器,它使用 AES-NI 集中的指令

二、调用oneMKL随机数算法;

这里可以使用的随机数算法很多,参考网址,如下:

单精度双精度描述
vsRngUniformvdRngUniform生成具有均匀分布的随机数
vsRngGaussianvdRngGaussian生成正态分布的随机数
vsRngGaussianMVvdRngGaussianMV从多元正态分布生成随机数
vsRngExponentialvdRngExponential生成指数分布的随机数
vsRngLaplacevdRngLaplace生成指数分布的随机数
vsRngWeibullvdRngWeibull分布随机数
vsRngCauchyvdRngCauchy生成柯西分布随机值
vsRngRayleighvdRngRayleigh生成柯西分布随机值
vsRngLognormalvdRngLognormal生成对数正态分布的随机数
vsRngGumbelvdRngGumbel分布随机值
vsRngGammavdRngGamma分布随机值
vsRngBetavdRngBeta生成 beta 分布随机值
vsRngChiSquarevdRngChiSquare生成卡方分布随机值

本博主使用生成随高斯分布的随机数函数vsRngGaussianvdRngGaussian,参考网址,其语法如下:

1.核心代码

#include <mkl.h>

auto status = vsRngGaussian( method, stream, n, r, a, sigma );
auto status = vdRngGaussian( method, stream, n, r, a, sigma );

2、算法描述

该函数生成具有正态(高斯)分布的随机数,平均值为a,标准差sigma

3、输入参数

名字类型描述
methodconst MKL_INT随机数生成方法,可选方法参考如下method 参数的定义
streamVSLStreamStatePtr指向流状态结构的指针
nconst MKL_INT要生成的随机值的数量。
rfloat* for vsRngGaussian double* for vdRngGaussian存放随机数的数组
aconst float for vsRngGaussian const double for vdRngGaussian平均值
sigmaconst float for vsRngGaussian const double for vdRngGaussian标准差

4、输出参数

该函数运行完后随机数结果都保存在r中,可以通过其返回的值status进行判断,判断如下:

名字描述
VSL_ERROR_OK指示无错误,执行成功
VSL_STATUS_OK指示无错误,执行成功
VSL_ERROR_NULL_PTRstream无效,stream是空指针
VSL_RNG_ERROR_BAD_STREAMstream无效
VSL_RNG_ERROR_BAD_UPDATE不支持非确定性随机数生成器
VSL_RNG_ERROR_NO_NUMBERS抽象 BRNG 的回调函数返回 0 作为缓冲区中更新条目的数量
VSL_RNG_ERROR_QRNG_PERIOD_ELAPSED已超过生成的周期
VSL_RNG_ERROR_QRNG_PERIOD_ELAPSED使用非确定性随机数生成器生成随机数的重试次数超过阈值
VSL_RNG_ERROR_ARS5_NOT_SUPPORTED运行应用程序的 CPU 不支持 ARS-5 随机数生成器

5、method 参数的定义

名字描述
VSL_RNG_METHOD_GAUSSIAN_BOXMULLER根据公式通过一对均匀分布的数u1和u2生成正态分布的随机数x x = − 2 ln ⁡ u 1 sin ⁡ 2 π u 2 x=\sqrt{-2 \ln u_{1}} \sin 2 \pi u_{2} x=2lnu1 sin2πu2
VSL_RNG_METHOD_GAUSSIAN_BOXMULLER2通过一对均匀分布数u1和u2根据以下公式生成正态分布随机数x1和x2 x 1 = − 2 ln ⁡ u 1 sin ⁡ 2 π u 2 x 2 = − 2 ln ⁡ u 1 cos ⁡ 2 π u 2 . \begin{array}{l}x_{1}=\sqrt{-2 \ln u_{1}} \sin 2 \pi u_{2}\\x_{2}=\sqrt{-2 \ln u_{1}} \cos 2 \pi u_{2}\end{array}. x1=2lnu1 sin2πu2x2=2lnu1 cos2πu2.
VSL_RNG_METHOD_GAUSSIAN_ICDF逆累积分布函数法

三、删除随机数流(控制台)

一个优秀的C/C++要学会自己管理内存,要记得手动删除不必要的内存。

1.核心代码

#include <mkl.h>

auto status = vslDeleteStream( &stream );

2、算法描述

该函数删除随机流。

3、输入参数

名字类型描述
streamVSLStreamStatePtr指向流状态结构的指针

4、输出参数

该函数运行完后可以通过其返回的值status进行判断,判断如下:

名字描述
VSL_ERROR_OK指示无错误,执行成功
VSL_STATUS_OK指示无错误,执行成功
VSL_ERROR_NULL_PTRstream无效,stream是空指针
VSL_RNG_ERROR_BAD_STREAMstream无效

四、使用实例

1.核心代码

#include <iostream>
#include <mkl.h>
#include <vector>
#include <numeric>
#include <algorithm>
#include <stdio.h>
#include <string>
using namespace std;

int main()
{
	// 定义大小
	int size = 2048;
	MKL_LONG sizeN[2] = { size, size };
	int size2 = size * size;

	// 开辟数组空间
	vector<float> randnData(size2, 0);
	float* pRandnData = &randnData[0];  // 创建索引指针

	VSLStreamStatePtr stream;
	int seed = 666;  // 随机种子
	float randn_aver = 0, randn_std = 1;
	auto status = vslNewStream(&stream, VSL_BRNG_MT19937, seed);

	// vsRngGaussian 是单精度 vdRngGaussian 是双精度  vsRngGaussian 生成正态分布的随机数
	status = vsRngGaussian(VSL_RNG_METHOD_GAUSSIAN_ICDF, stream, size2, pRandnData, randn_aver, randn_std);
	status = vslDeleteStream(&stream);

	// 最大值  
	double max_value = *max_element(randnData.begin(), randnData.end());
	// 最小值  
	double min_value = *min_element(randnData.begin(), randnData.end());
	// 均值  
	double mean = accumulate(randnData.begin(), randnData.end(), 0.0) / randnData.size();
	// 方差  
	double variance = 0.0;
	for (int i = 0; i < randnData.size(); i++)
	{
		variance += pow(randnData[i] - mean, 2);
	}
	variance /= randnData.size();

	// 标准差  
	double std_dev = sqrt(variance);

	cout << "=================数据输出信息=====================" << endl;
	printf("生成正态分布的随机数 均值 %f 方差 %f\n", randn_aver, randn_std);
	printf("前5位数:%f %f %f %f %f \n", randnData[0], randnData[1], randnData[2], randnData[3], randnData[4]);
	printf("后5位数:%f %f %f %f %f \n", randnData[size2 - 1], randnData[size2 - 2], randnData[size2 - 3], randnData[size2 - 4], randnData[size2 - 5]);
	cout << "最大值: " << max_value << endl
		<< "最小值:" << min_value << endl
		<< "均值:" << mean << endl
		<< "方差:" << variance << endl
		<< "标准差 :" << std_dev << endl;

	return 0;
}

2、算法描述

上述代码使用vsRngGaussian生成了单精度的正态(高斯)分布的随机数,平均值为a=0,标准差sigma=1,并使用vector的一些函数统计了生成后的随机数randnData的信息。

3、运行结果

运行结果如下图所示:
随机数生成结果
通过上图运行结果可以看出,生成的随机数符合预先设定的平均值为a=0,标准差sigma=1


最后

欢迎大家积极提出宝贵的建议 !
支持一下,收藏!点赞!!加星星!!!
爱你哦

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值