在Blob进阶1基础上更改代码:
#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()函数修改数值
for(int i=0;i<a.count();i++)
{
p[i]=i;
}
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范数(平方和)
}
编译出错:
编译改为:
g++ -o app blob_demo.cpp -I/home/hk/Documents/caffe-master/include/ -D CPU_ONLY -I/home/hk/Documents/caffe-master/.build_release/src/ -L/home/hk/Documents/caffe-master/build/lib/ -lcaffe -lglog -Wl,--rpath-link=.
结果:
w出错,代码敲错了,改后结果:
发现blob存储和c++高维数组一致 想想Blob进阶1 : Blob在内存中表示四维数组,维度由高到底为(num_,channels_,height_,width)