0

基本上我有一个需要一些随机性的 ufunc。为了使 ufunc 尽可能地重现，我想为此使用 numpy 的随机数生成器；基本上是因为这样设置种子会更直观。

但是我找不到 numpy.random C-API 的文档（有吗？）。

我目前的方法是这样的：

#include <numpy/random.h> // does not exist
...

static void
ufunc_M( char ** args
       , npy_intp * dimensions
       , npy_intp * steps
       , void * data)
{
    basic_gate_argument_t argument = *((basic_gate_argument_t *) data);
    PYQCS_GATE_GENERIC_SETUP;
    npy_intp i;


    npy_double amplitude_1 = 0;

    for(i = 0; i < ndim; i++)
    {
        if(i & (1 << argument.act))
        {
            amplitude_1 += qm_in[i].real * qm_in[i].real;
            amplitude_1 += qm_in[i].imag * qm_in[i].imag;
        }
    }

    npy_double rand = random_uniform(0, 1); // does not exist

    ...

    *measured_out = 1 << argument.act;
}

...
4

1 回答 1

0

事实证明：不能numpy.random直接使用库来做到这一点。在搜索了 numpy 标头一段时间后，我发现了这一点。随机库只是不导出。

我的解决方法是将可调用的 python 传递给 ufunc：

static void
ufunc_M( char ** args
       , npy_intp * dimensions
       , npy_intp * steps
       , void * data)
{
    basic_gate_argument_t argument = *((basic_gate_argument_t *) data);
    PYQCS_GATE_GENERIC_SETUP;
    npy_intp i;

    ...

    npy_double randr;
    //==================================================//
    // Get some random value. I do not like the way this
    // is done but it seems like there is no better way.
    PyObject * random_result = PyObject_CallFunctionObjArgs(argument.rng, NULL);

    if(!PyFloat_Check(random_result))
    {
        randr = 0;
    }
    else
    {
        randr = PyFloat_AsDouble(random_result);
    }
    Py_DECREF(random_result);
    //==================================================//

    ...
}

在构造 ufunc 时，必须检查传递的对象是否为可调用对象：

static int
BasicGate_init
    ( BasicGate * self
    , PyObject * args)
{
    char type;

    if(!PyArg_ParseTuple(args, "ClldO"
                , &type
                , &(self->argument.act)
                , &(self->argument.control)
                , &(self->argument.r)
                , &(self->argument.rng))
            )
    {
        return -1;
    }

    if(!PyCallable_Check(self->argument.rng))
    {
        PyErr_SetString(PyExc_TypeError, "random (5th argument) must be a callable (returning float)");
        return -1;
    }

  ...

        case 'M':
        {
            self->ufunc = PyUFunc_FromFuncAndDataAndSignature(
                ufunc_M_funcs // func
                , self->data // data
                , ufunc_types //types
                , 1 // ntypes
                , 2 // nin
                , 3 // nout
                , PyUFunc_None // identity
                , "M_function" // name
                , "Computes the M (Measurement) gate on a state." // doc
                , 0 // unused
                , "(n),(m)->(n),(m),()"); 

    ...
}

然而，Ufuncs 不会失败，所以当传递的函数没有返回PyFloats 时，ufunc 会产生废话。

于 2019-09-11T07:55:06.807 回答