Browse Source

Added get_tensor_by_name to the Graph class

tags/v0.9
Juanu Haedo 6 years ago
parent
commit
31a106dc23
1 changed files with 13 additions and 1 deletions
  1. +13
    -1
      src/TensorFlowNET.Core/Graphs/Graph.cs

+ 13
- 1
src/TensorFlowNET.Core/Graphs/Graph.cs View File

@@ -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__()
{
}


Loading…
Cancel
Save