通常、ウェイトやバイアスを含む「tensor_content」を取得するには？

Question

私はtensorflowフレームワークを使用するC++プログラムを作成しようとしています。今私は重みにアクセスし、それをファイルに書き込もうとしています。私は以下に示すようにtensor_contentに記憶される重みにアクセスすることができません

int node_count = graph_def.node_size(); 
for (int i = 0; i < node_count; i++) 
{ 
     auto n = graph_def.node(i); 
     cout<<"Names : "<< n.name() <<endl; 

} 

、

node { 
    name: "Weight" 
    op: "Const" 
    attr { 
    key: "dtype" 
    value { 
     type: DT_FLOAT 
    } 
    } 
    attr { 
    key: "value" 
    value { 
     tensor { 
     dtype: DT_FLOAT 
     tensor_shape { 
      dim { 
      size: 8 
      } 
      dim { 
      size: 2 
      } 
     } 
     tensor_content: "\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377\000\000\300\377" 
     } 
    } 
    } 
} 

ありがとう予め、以下のコードによってノードの名前にアクセスすることができる午前。

Answer 1

以下のコードでは、配列変数にはtensor_contentが含まれています。あなたは彼らのインデックスを与えることによってそれらにアクセスすることができます。

std::vector<Tensor> output; 
std::vector<string> vNames; 

for (int i = 0; i < node_count; i++) 
{ 
     auto n = graph_def.node(i); 
     if ((has_suffix(n.name(), "/read")) || (has_suffix(n.name(), "_w")) || (has_suffix(n.name(), "_b"))) 
     { 
       vNames.push_back(n.name()); 
     } 
} 
Status run_status = session->Run({}, vNames, {}, &output); 
auto array = output[i].tensor<float, 4 >();