Blob进阶3

来源:互联网 发布:郑州学历网络教育 编辑:程序博客网 时间:2024/06/13 23:36

Update操作将data与diff融合,是权值更新最终实施者 data=data-diff


代码:

#include<vector>#include<iostream>#include<caffe/blob.hpp>using namespace caffe;using namespace std;int main(void){ Blob<float> a; cout<<"Size: "<< a.shape_string()<<endl; a.Reshape(1,2,3,4);   //调用Reshape赋值 cout<<"Size : "<<a.shape_string()<<endl;  float *p=a.mutable_cpu_data(); //mutable_cpu_data()函数修改数值  float *q=a.mutable_cpu_diff(); //diff权值增量,源码频繁出现 for(int i=0;i<a.count();i++)  {    p[i]=i;      //把data初始化为0,1,2,3    q[i]=a.count()-1-i;//把diff初始化为23,22,21.。。      }   }  a.Update(); //此Update操作将data与diff融合,是权值更新最终实施者 data=data-diff  for(int u=0;u<a.num();u++)     {       for(int v=0;v<a.channels();v++)         {           for(int w=0;w<a.height();w++)             {               for(int x=0;x<a.width();x++)                  {                    cout<<"a["<<u<<"]["<<v<<"]["<<w<<"]["<<x<<"]="<<a.data_at(u,v,w,x)<<endl;    }  }}     }         cout<<"ASUM="<<a.asum_data()<<endl;  //计算所以元素的L1范数 (绝对值之和)         cout<<"SUMSQ="<<a.sumsq_data()<<endl;//计算所以元素的L2范数(平方和)        }

编译运行同Blob2

结果:


0 0
原创粉丝点击