From 31a106dc23083207c3082a80fe726505edb39d18 Mon Sep 17 00:00:00 2001
From: Juanu Haedo <juanu@juanu.com.ar>
Date: Wed, 1 May 2019 19:48:30 -0300
Subject: [PATCH] Added get_tensor_by_name to the Graph class

---
 src/TensorFlowNET.Core/Graphs/Graph.cs | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/TensorFlowNET.Core/Graphs/Graph.cs b/src/TensorFlowNET.Core/Graphs/Graph.cs
index f1a33371..adfbfcb9 100644
--- a/src/TensorFlowNET.Core/Graphs/Graph.cs
+++ b/src/TensorFlowNET.Core/Graphs/Graph.cs
@@ -388,7 +388,7 @@ namespace Tensorflow
                 var handle = return_oper_handle.node + Marshal.SizeOf<TF_Operation>() * i;
                 return_opers[i] = new Operation(*(IntPtr*)handle);
             }
-
+            
             return return_opers;
         }
 
@@ -439,6 +439,18 @@ namespace Tensorflow
             c_api.TF_DeleteGraph(_handle);
         }
 
+        /// <summary>
+        /// Returns the <see cref="Tensor"/> with the given <paramref name="name"/>.
+        /// This method may be called concurrently from multiple threads.
+        /// </summary>
+        /// <param name="name">The name of the `Tensor` to return.</param>
+        /// <exception cref="KeyError">If <paramref name="name"/> does not correspond to a tensor in this graph.</exception>
+        /// <returns>The `Tensor` with the given <paramref name="name"/>.</returns>
+        public ITensorOrOperation get_tensor_by_name(string name)
+        {
+            return this.as_graph_element(name, allow_tensor: true, allow_operation: false);
+        }
+
         public void __enter__()
         {
         }