caffe数据结构(一): Blob

Blob是一个模板类, 其对象在内存中表示4维数组(num_, channels_,height_,width_).

在src/caffe/proto/caffe.proto中与Blob相关的数据结构(BlobShape 和BlobProto):

syntax = "proto2";

package caffe;

// Specifies the shape (dimensions) of a Blob.该结构描述Blob 的形状信息
message BlobShape {
  repeated int64 dim = 1 [packed = true];
}

//该结构描述Blob 在磁盘中序列化后的状态
message BlobProto { 
  optional BlobShape shape = 7; //可选,包括一个Blobshape对象
  repeated float data = 5 [packed = true]; //若干浮点元素,存储数据或权值, 元素数目由shape或(num,channel,height,width)确定
  repeated float diff = 6 [packed = true]; //存储增量信息,维度与data一致
  repeated double double_data = 8 [packed = true]; //与data并列,类型为double
  repeated double double_diff = 9 [packed = true];

  // 4D dimensions -- deprecated.  Use "shape" instead. 推荐使用shape
  optional int32 num = 1 [default = 0];
  optional int32 channels = 2 [default = 0];
  optional int32 height = 3 [default = 0];
  optional int32 width = 4 [default = 0];
}

Blob模板类的声明和实现:

//blob.hpp
#ifndef CAFFE_BLOB_HPP_
#define CAFFE_BLOB_HPP_

#include <algorithm>
#include <string>
#include <vector>

#include "caffe/common.hpp"
#include "caffe/proto/caffe.pb.h"//由protoc生成的头文件，声明了BlobProto/BlobShape等遵循caffe.proto协议的数据结构
#include "caffe/syncedmem.hpp" //CPU/GPU共享内存类，用于数据同步
//SyncedMemory类是一个CPU/GPU共享内存类,负责分配内存,以及主机和设备间的同步.

const int kMaxBlobAxes = 32;//Blob最大维数目

namespace caffe {
template <typename Dtype>
class Blob { //模板类Blob封装了SyncedMemory类
 public:
  Blob()
       : data_(), diff_(), count_(0), capacity_(0) {}//默认构造函数

  /// @brief Deprecated; use <code>Blob(const vecto