fvar_op7.cpp

Go to the documentation of this file.

#include "fvar.hpp"

dvariable& square(const prevariable& variable)
{
  dvariable* RETURN_PTR = gradient_structure::next_RETURN_PTR();

  double& x = value(variable);
  RETURN_PTR->v->x = x * x;

  gradient_structure::GRAD_STACK1->set_gradient_stack(default_evaluation2,
     &(RETURN_PTR->v->x), &(variable.v->x), 2 * x );

  return *RETURN_PTR;
}
dvar_matrix square(const dvar_matrix& matrix)
{
  dvar_matrix results;
  results.allocate(matrix);

  int min = matrix.rowmin();
  int max = matrix.rowmax();
  const dvar_vector* pmatrixi = &matrix(min);
  dvar_vector* presultsi = &results(min);
  for (int i = min; i <= max; ++i)
  {
    *presultsi = square(*pmatrixi);

    ++pmatrixi;
    ++presultsi;
  }
  return results;
}