伯努利分布函数

生成符合伯努利分布（bernoulli distribution）的随机数向量。

接口定义

int viRngBernoulli (const int method, VSLStreamStatePtr stream, const SizeType n, float *r, const double p);

概率密度函数

$\text{[math]}$

参数

参数名	类型	描述	输入/输出
method	int	随机数生成模式。 VSL_RNG_METHOD_BERNOULLI_ICDF：逆累积分布函数法模式	输入
stream	VSLStreamStatePtr类型	随机数流。	输入
n	SizeType	输入输出的长度。	输入
r	单精度浮点类型指针	指向结果的指针。	输出
p	double	伯努利分布的成功概率	输入

依赖

#include "krng.h"

示例

#include <stdio.h>
#include <stdlib.h>
#include "krng.h"
int main()
{
   /* initialize stream with given BRNG type and seed */
    VSLStreamStatePtr stream;
    unsigned seed = 42;
    int errcode = vslNewStream(&stream, VSL_BRNG_MCG59, seed);
    if (errcode != VSL_STATUS_OK) {
        fprintf(stderr, "Failure in newstream\n");
        return 0;
    }

    Sizetype n = 10;
    float *r = (float *)malloc(sizeof(float) * n);
    if (r == NULL) {
        fprintf(stderr, "Failure in malloc\n");
        return 0;
    }

    if (viRngBernoulli(VSL_RNG_METHOD_BERNOULLI_ICDF, stream, n, r, 0.5)) {
        fprintf(stderr, "Failure in (viRngBernoulli\n");
        goto out;
    }

    /* deinitialize the stream */
    errcode = vslDeleteStream(&stream);
    if (errcode != VSL_STATUS_OK) {
        fprintf(stderr, "Failure in deleting stream\n");
        goto out;
    }

out:
    free(r);
    return 0;
}

父主题： 高性能函数