Caffe blob reshape

最新推荐文章于 2020-12-25 17:35:32 发布

visionfans

最新推荐文章于 2020-12-25 17:35:32 发布

阅读量5.2k

点赞数 1

分类专栏： Caffe

本文链接：https://blog.csdn.net/visionfans/article/details/48400147

版权

Caffe 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

Release original memory and hold new:


template <typename Dtype>
void Blob<Dtype>::Reshape(const vector<int>& shape) {
  CHECK_LE(shape.size(), kMaxBlobAxes);
  count_ = 1;
  shape_.resize(shape.size());
  for (int i = 0; i < shape.size(); ++i) {
    CHECK_GE(shape[i], 0);
    CHECK_LE(shape[i], INT_MAX / count_) << "blob size exceeds INT_MAX";
    count_ *= shape[i];
    shape_[i] = shape[i];
  }
  if (count_ > capacity_) {
    capacity_ = count_;
    data_.reset(new SyncedMemory(capacity_ * sizeof(Dtype)));
    diff_.reset(new SyncedMemory(capacity_ * sizeof(Dtype)));
  }
}