feed_dict_tensor in BaseSession._run

Oceania2018 · Oceania2018 · commit 5f595eb5eb8f · 2018-12-19T17:44:37.000-06:00
diff --git a/src/TensorFlowNET.Core/BaseSession.cs b/src/TensorFlowNET.Core/BaseSession.cs
@@ -1,4 +1,5 @@
-﻿using System;
+﻿using NumSharp.Core;
+using System;
 using System.Collections.Generic;
 using System.Text;
 
@@ -44,6 +45,24 @@ public virtual byte[] run(Tensor fetches, Dictionary<Tensor, object> feed_dict =
 
         private unsafe byte[] _run(Tensor fetches, Dictionary<Tensor, object> feed_dict = null)
         {
+            var feed_dict_tensor = new Dictionary<Tensor, NDArray>();
+
+            if (feed_dict != null)
+            {
+                NDArray np_val = null;
+                foreach (var feed in feed_dict)
+                {
+                    switch (feed.Value)
+                    {
+                        case float value:
+                            np_val = np.asarray(value);
+                            break;
+                    }
+
+                    feed_dict_tensor[feed.Key] = np_val;
+                }
+            }
+            
             var status = new Status();
 
             c_api.TF_SessionRun(_session,
diff --git a/src/TensorFlowNET.Core/Tensor.cs b/src/TensorFlowNET.Core/Tensor.cs
@@ -13,6 +13,8 @@ public class Tensor
         private DataType _dtype;
         public DataType dtype => _dtype;
 
+        public string name;
+
         public Tensor(Operation op, int value_index, DataType dtype)
         {
             _op = op;

Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,8 @@ public class Tensor`
`13`	`13`	`private DataType _dtype;`
`14`	`14`	`public DataType dtype => _dtype;`
`15`	`15`
	`16`	`+ public string name;`
	`17`	`+`
`16`	`18`	`public Tensor(Operation op, int value_index, DataType dtype)`
`17`	`19`	`{`
`18`	`20`	`_op = op;`