Caffe2填坑系列(8)----得到Op输入和输出blob的个数的两种方法

构造函数中,任何一个Op都是公有继承Operator,构造函数通常是web C++ :Operator<Context>(def,ws), … 能够看到每个构造函数里都有一个OperatorDef的对象def,OperatorDef在protot里进行定义,主要时组着里一个Op的一些信息,咱们能够调用def.input_size()或def.output_size()得到输入输出的blob数量.s
相关文章
相关标签/搜索