caffe源碼解析:Blob<Dtype>::Reshape

Blob<Dtype>::Reshape(const vector<int>& shape) 功能:爲blob分配內容空間 該函數一般在開闢空間內存時調用,如輸入的data層,InputLayer<Dtype>::LayerSetUp中調用。 一般shape是(num, channel, width, height), 容量count = num*channel*width*height 是空間的
相關文章
相關標籤/搜索