caffe源码解析 — blob.cpp

主要参考：linger

Reshape(const int num, const int channels, const int height, const int width)
功能：改变一个blob的大小
步骤：1.读入num_，channels_，height_，width_的大小
2.计算count_：count_ = num_ * channels_ * height_ * width_;
3.如果count_不为0，则重新为data_和diff_分配一块空间
如果count为0，则都初始化为NULL
输入：num，channels，height，width
输出：无

ReshapeLike(const Blob<Dtype>& other)
功能：为data_和diff_ 重新分配一块空间，大小和另一个blob的一样
输入：Bolb类型的other
输出：无

Blob<Dtype>::Blob(const int num, const int channels, const int height, const int width)
功能：简单的构造函数
输入：num，channels，height，width

set_cpu_data(Dtype* data)
功能：改变CPU的数据

const Dtype* Blob<Dtype>::cpu_data()
const Dtype* Blob<Dtype>::cpu_diff()
const Dtype* Blob<Dtype>::gpu_data()
const Dtype* Blob<Dtype>::gpu_diff()

功能：以上四个函数，前两个调用to_cpu(),返回cpu_ptr；第一个对于data对象，第二个对于diff对象
后两个 调用to_gpu(),返回gpu_ptr；第一个对于data对象，第二个对于diff对象

Dtype* Blob<Dtype>::mutable_cpu_data()
Dtype* Blob<Dtype>::mutable_gpu_data()
Dtype* Blob<Dtype>::mutable_cpu_diff()
Dtype* Blob<Dtype>::mutable_gpu_diff()

功能：下面是gpu_data和mutable_gpu_data函数的对比，可以看出第二个函数
只是多改变了head_的值。

const void* SyncedMemory::gpu_data() {
  to_gpu();
  return (const void*)gpu_ptr_;
}

void* SyncedMemory::mutable_gpu_data() {
  to_gpu();
  head_ = HEAD_AT_GPU;
  return gpu_ptr_;
}

void Blob<Dtype>::ShareData(const Blob& other)
void Blob<Dtype>::ShareDiff(const Blob& other)

功能：从other的blob复制data和diff的值

Update()
功能：更新data_的数据，就是减去diff_的数据。
步骤：1.判断blob的位置（HEAD_AT_CPU/HEAD_AT_GPU/SYNCED/UNINITIALIZED）
1）调用caffe_axpy：在math_functions.cpp可以找到该函数的实现，其实这函数也是封装了mkl的函数。这里调用是为了实现了两个向量的减法。
2）调用caffe_gpu_axpy：在math_functions.cpp可以找到该函数的实现，其实这函数也是封装了cublas的函数。这里调用是为了实现了两个向量的减法。

CopyFrom(const Blob& source, bool copy_diff, bool reshape)
功能：从source拷贝数据。copy_diff作为标志来区分是拷贝data还是拷贝diff。
步骤：1.如果是GPU：
如果是拷贝diff：调用cudaMemcpy函数将source的diff拷贝过来
否则拷贝data
2.如果是CPU：
如果是拷贝diff：调用memcpy函数将source的diff拷贝过来
否则拷贝data

FromProto(const BlobProto& proto)
功能：从proto读数据进来，其实就是反序列化
步骤：1.先把blob的大小改变一下
2.得到cpu中数据的地址
3.用proto中的data覆盖blob中的data
4.用proto中的diff覆盖blob中的diff

ToProto(BlobProto* proto, bool write_diff)
功能：把blob数据保存到proto中