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

void MyLossLayer::Forward_cpu(const vector& bottom,
		const vector& top) {

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

	LOG(INFO)<<"=====>>>>>MyLossLayer";
	for(int i=0;i//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)

你可能感兴趣的:(caffe 打印出Forward 函数输入输出的shape)