首页 > 其他分享 >caffe 打印出Forward 函数输入输出的shape

caffe 打印出Forward 函数输入输出的shape

时间:2023-01-04 14:34:35浏览次数:44  
标签:string bottom int MyLossLayer shape caffe Forward

void MyLossLayer<Ftype, Btype>::Forward_cpu(const vector<Blob*>& bottom,
		const vector<Blob*>& top) {

	//	bottom[0]: "depth" 1x1x36x64
	//	bottom[1]: "data" for loss reshape
	//	bottom[2]: "label"

	LOG(INFO)<<"=====>>>>>MyLossLayer";
	for(int i=0;i<bottom.size();i++)
	{
		LOG(INFO)<<"bottom["<<i<<"]="<<bottom[i]->shape_string();
	}
	for(int i=0;i<top.size();i++)
	{
		LOG(INFO)<<"top["<<i<<"]="<<top[i]->shape_string();
	}
...
//bottom[0]=14 1 48 144 (96768)  conv_depth_1x1
//bottom[1]=14 3 192 576 (4644864)  data
// bottom[2]=1 1 84 35 (2940)  label
// top[0]=(1)

标签:string,bottom,int,MyLossLayer,shape,caffe,Forward
From: https://www.cnblogs.com/yanghailin/p/17024712.html

相关文章