Prev Next atomic_four_vector_add.cpp

@(@\newcommand{\W}[1]{ \; #1 \; } \newcommand{\R}[1]{ {\rm #1} } \newcommand{\B}[1]{ {\bf #1} } \newcommand{\D}[2]{ \frac{\partial #1}{\partial #2} } \newcommand{\DD}[3]{ \frac{\partial^2 #1}{\partial #2 \partial #3} } \newcommand{\Dpow}[2]{ \frac{\partial^{#1}}{\partial {#2}^{#1}} } \newcommand{\dpow}[2]{ \frac{ {\rm d}^{#1}}{{\rm d}\, {#2}^{#1}} }@)@This is cppad-20221105 documentation. Here is a link to its current documentation .
Atomic Vector Addition Example

f(u, v, w)
For this example, @(@ f : \B{R}^{3m} \rightarrow \B{R}^m @)@ is defined by @(@ f(u, v, w) = u + v + w @)@. where u , v , and w are in @(@ \B{R}^m @)@.

g(u, v, w)
For this example @(@ g : \B{R}^{3m} \rightarrow \B{R}^m @)@ is defined by @(@ g_i (u, v, w) = \partial_{v[i]} f_i (u, v, w) @)@

Source

# include <cppad/cppad.hpp>
# include <cppad/example/atomic_four/vector/vector.hpp>
bool add(void)
{   bool ok = true;
    using CppAD::NearEqual;
    using CppAD::AD;
    double eps99 = 99.0 * CppAD::numeric_limits<double>::epsilon();
    //
    // vec_op
    // atomic vector_op object
    CppAD::atomic_vector<double> vec_op("atomic_vector");
    //
    // m
    // size of u, v, and w
    size_t m = 5;
    //
    // add_op
    typedef CppAD::atomic_vector<double>::op_enum_t op_enum_t;
    op_enum_t add_op = CppAD::atomic_vector<double>::add_enum;
    // -----------------------------------------------------------------------
    // Record f(u, v, w) = u + v + w
    // -----------------------------------------------------------------------
    // Independent variable vector
    CPPAD_TESTVECTOR( CppAD::AD<double> ) auvw(3 * m);
    for(size_t j = 0; j < 3 * m; ++j)
        auvw[j] = AD<double>(1 + j);
    CppAD::Independent(auvw);
    //
    // au, av, aw
    CPPAD_TESTVECTOR( CppAD::AD<double> ) au(m), av(m), aw(m);
    for(size_t i = 0; i < m; ++i)
    {   au[i] = auvw[0 * m + i];
        av[i] = auvw[1 * m + i];
        aw[i] = auvw[2 * m + i];
    }
    //
    // ax = (au, av)
    CPPAD_TESTVECTOR( CppAD::AD<double> ) ax(2 * m);
    for(size_t i = 0; i < m; ++i)
    {   ax[i]     = au[i];
        ax[m + i] = av[i];
    }
    //
    // ay = u + v
    CPPAD_TESTVECTOR( CppAD::AD<double> ) ay(m);
    vec_op(add_op, ax, ay);
    //
    // ax = (ay, aw)
    for(size_t i = 0; i < m; ++i)
    {   ax[i]     = ay[i];
        ax[m + i] = aw[i];
    }
    //
    // az = ay + w
    CPPAD_TESTVECTOR( CppAD::AD<double> ) az(m);
    vec_op(add_op, ax, az);
    //
    // f
    CppAD::ADFun<double> f(auvw, az);
    // -----------------------------------------------------------------------
    // check forward mode on f
    // -----------------------------------------------------------------------
    //
    // uvw, duvw
    CPPAD_TESTVECTOR(double) uvw(3 * m), duvw(3 * m);
    for(size_t j = 0; j < 3 * m; ++j)
    {   uvw[j]  = double(1 + j);
        duvw[j] = double(j);
    }
    //
    // z, dz
    CPPAD_TESTVECTOR(double) z(m), dz(m);
    z  = f.Forward(0, uvw);
    dz = f.Forward(1, duvw);
    //
    // ok
    for(size_t i = 0; i < m; ++i)
    {   double check_z  = uvw[0 * m + i] + uvw[1 * m + i] + uvw[2 * m + i];
        ok             &= NearEqual( z[i] ,  check_z,  eps99, eps99);
        double check_dz = double( (0 * m + i)  + (1 * m + i) + (2 * m + i) );
        ok             &= NearEqual( dz[i] ,  check_dz,  eps99, eps99);
    }
    // -----------------------------------------------------------------------
    // check reverse mode on f
    // -----------------------------------------------------------------------
    //
    // weight
    CPPAD_TESTVECTOR(double) weight(m);
    for(size_t i = 0; i < m; ++i)
        weight[i] = 1.0;
    //
    // dweight
    CPPAD_TESTVECTOR(double) dweight(3 * m);
    f.Forward(0, uvw);
    dweight = f.Reverse(1, weight);
    //
    // ok
    for(size_t j = 0; j < 3 * m; ++j)
    {   double check  = 1.0;
        ok           &= NearEqual(dweight[j], check, eps99, eps99);
    }
    // -----------------------------------------------------------------------
    // Record g_i (u, v, w) = \partial d/dv[i] f_i (u , v , w)
    // -----------------------------------------------------------------------
    //
    // af
    CppAD::ADFun< AD<double>, double > af = f.base2ad();
    //
    // auvw
    CppAD::Independent(auvw);
    //
    // aduvw
    CPPAD_TESTVECTOR( AD<double> ) aduvw(3 * m);
    for(size_t i = 0; i < m; ++i)
    {   aduvw[0 * m + i]  = 0.0; // du[i]
        aduvw[1 * m + i]  = 1.0; // dv[i]
        aduvw[2 * m + i]  = 0.0; // dw[i]
    }
    //
    // az
    // use the fact that d_v[i] f_k (u, v, w) is zero when i != k
    af.Forward(0, auvw);
    az = af.Forward(1, aduvw);
    CppAD::ADFun<double> g(auvw, az);
    // -----------------------------------------------------------------------
    // Record h (u, v, w) = sum f_i^(1) (u , v , w)
    // -----------------------------------------------------------------------
    //
    // auvw
    CppAD::Independent(auvw);
    //
    // aweight
    CPPAD_TESTVECTOR( AD<double> ) aweight(m);
    for(size_t i = 0; i < m; ++i)
        aweight[i] = 1.0;
    //
    // az
    CPPAD_TESTVECTOR( AD<double> ) adweight(3 * m);
    af.Forward(0, auvw);
    az = af.Reverse(1, aweight);
    CppAD::ADFun<double> h(auvw, az);
    // -----------------------------------------------------------------------
    // check forward mode on g
    // -----------------------------------------------------------------------
    //
    // z
    z = g.Forward(0, uvw);
    //
    // ok
    for(size_t i = 0; i < m; ++i)
    {   double check_z  = 1.0;
        ok             &= NearEqual( z[i] ,  check_z,  eps99, eps99);
    }
    // -----------------------------------------------------------------------
    // check forward mode on h
    // -----------------------------------------------------------------------
    //
    // z
    z = h.Forward(0, uvw);
    //
    // ok
    for(size_t j = 0; j < 3 * m; ++j)
    {   double check_z  = 1.0;
        ok             &= NearEqual( z[j] ,  check_z,  eps99, eps99);
    }
    return ok;
}

Input File: example/atomic_four/vector/add.cpp