Browse Source

get tensor index as per PR comment

tags/v0.30
Chris Fairclough Esther Hu 5 years ago
parent
commit
ddbe47e2de
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