File: base2vec_ad.cpp

package info (click to toggle)
cppad 2025.00.00.2-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 11,552 kB
  • sloc: cpp: 112,594; sh: 5,972; ansic: 179; python: 71; sed: 12; makefile: 10
file content (88 lines) | stat: -rw-r--r-- 2,017 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
// SPDX-FileCopyrightText: Bradley M. Bell <bradbell@seanet.com>
// SPDX-FileContributor: 2003-22 Bradley M. Bell
// ----------------------------------------------------------------------------

/*
{xrst_begin base2vec_ad.cpp}

Using base2ad and VecAD together: Example and Test
##################################################

Purpose
*******
This example demonstrate that :ref:`base2ad-name` does not convert
:ref:`VecAD-name` operations as might be expected.
To be specific, this indexing into a VecAD object gets fixed
when zero order forward mode is run for a ``base2ad`` result; e.g.
*af* below.

{xrst_literal
   // BEGIN C++
   // END C++
}

{xrst_end base2vec_ad.cpp}
*/
// BEGIN C++
# include <cppad/cppad.hpp>

bool base2vec_ad(void)
{  using CppAD::vector;
   using CppAD::AD;
   //
   bool ok = true;
   //
   // f(x) = x[0] if 0 <= x[2] < 1
   //      = x[1] if 1 <= x[2] < 2
   vector< AD<double> > ax(3), ay(1);
   ax[0] = 3.0;
   ax[1] = 2.0;
   ax[2] = 1.0;
   Independent(ax);
   CppAD::VecAD<double> av(2);
   av[ AD<double>(0) ] = ax[0];
   av[ AD<double>(1) ] = ax[1];
   ay[0] = av[ ax[2] ];
   CppAD::ADFun<double> f(ax, ay);
   //
   // ok
   // value during recording of f
   ok &= ay[0] == ax[1];
   //
   // ok
   // f zero order forward mode, the VecAD commands give expected result
   vector<double> x(3), y(1);
   x[0] = 2.0;
   x[1] = 1.0;
   x[2] = 0.0;
   y = f.Forward(0, x);
   ok &= y[0] == x[0];
   //
   // af
   CppAD::ADFun< AD<double>, double > af = f.base2ad();
   //
   // g
   ax[0] = 3.0;
   ax[1] = 2.0;
   ax[2] = 1.0;
   Independent(ax);
   ay = af.Forward(0, ax);
   CppAD::ADFun<double> g(ax, ay);
   //
   // ok
   // value during recording of g
   ok &= ay[0] == ax[1];
   //
   // ok
   // g zero order forward mode
   // Note that this index does not change, but the value does
   x[0] = 5.0;
   x[1] = 4.0;
   x[2] = 0.0;
   y = g.Forward(0, x);
   ok &= y[0] == x[1];
   //
   return ok;
}
// END C++