difference gradient

#include <math.h> 
#include <stdlib.h>
#include <stdio.h>

void show_vector(float *A, int n) {
    for (int i = 0; i < n; i++) {
            printf("%2.5f \t", A[i]);
    }
    printf("\n");
}

float func( float* x )
{
    float result = 0.0;
    result = x[0] * x[0] * x[0] + x[1] *x[1];
    return result;
}

/*
  Calculating gradient of a function with different methods
*/
void forward()
{
   int n = 2;

   float x[] = { 0, 0};
   float pf[] = { 0, 0};
   float dx = 0.001;
   float f = func( x );


   for( int i = 0; i < n; i++ )
   {
       x[i] += dx;
       float newF = func( x );
       float diff = newF - f;
       pf[i] = diff/dx;
       x[i] -= dx; 
   }
   printf( "forward pf =  \n" );
   show_vector( pf, n );

}
void twoside()
{
   int n = 2;

   float x[] = { 0, 0};
   float pf[] = { 0, 0};
   float dx = 0.01;
   float f = func( x );


   for( int i = 0; i < n; i++ )
   {
       x[i] += dx;
       float F_forward = func( x );
       x[i] -= 2*dx; 
       float F_backward = func( x );
       x[i] += dx;
       float diff = F_forward - F_backward ;
       pf[i] = diff/(2*dx);
   }
   printf( "two side pf =  \n" );
   show_vector( pf, n );
}
void fourthorder()
{
   int n = 2;

   float x[] = { 0, 0};
   float pf[] = { 0, 0};
   float dx = 0.01;
   float f = func( x );


   for( int i = 0; i < n; i++ )
   {
       float x0 = x[i];
       x[i] = x0 + dx;
       float F_forward_1 = func( x );

       x[i] = x0 + 2*dx;
       float F_forward_2 = func( x );

       x[i] = x0 - dx; 
       float F_backward_1 = func( x );

       x[i] = x0 - 2*dx; 
       float F_backward_2 = func( x );
       x[i] = x0;

       float diff = 2* F_forward_2 + 16*F_forward_1 - 16 F_backward_1 - 2* F_backward_2 ;
       pf[i] = diff/(4*dx);
   }
   printf( "fourth order pf =  \n" );
   show_vector( pf, n );   
}
void main()
{
    forward();
    twoside();
    fourthorder();
    return;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值