Browse Source

get tensor index as per PR comment

pull/623/head
Chris Fairclough 5 years ago
parent
commit
e1c8662af8
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      src/TensorFlowNET.Core/Framework/importer.cs

+ 2
- 1
src/TensorFlowNET.Core/Framework/importer.cs View File

@@ -121,7 +121,8 @@ namespace Tensorflow

foreach(var input in input_map)
{
c_api.TF_ImportGraphDefOptionsAddInputMapping(options.Handle, input.Key, 0, input.Value._as_tf_output());
var (src_name, src_index) = _ParseTensorName(input.Key);
c_api.TF_ImportGraphDefOptionsAddInputMapping(options.Handle, src_name, src_index, input.Value._as_tf_output());
}

if (return_elements == null)


Loading…
Cancel
Save