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
相關文章
相關標籤/搜索