Browse Source

overload session.run for 4 tuples.

tags/v0.13
Oceania2018 5 years ago
parent
commit
1d9e2e57cd
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      src/TensorFlowNET.Core/Sessions/BaseSession.cs

+ 8
- 0
src/TensorFlowNET.Core/Sessions/BaseSession.cs View File

@@ -68,6 +68,14 @@ namespace Tensorflow
return _run(fetche, feed_dict)[0];
}

public virtual (NDArray, NDArray, NDArray, NDArray, NDArray) run(
(ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation) fetches,
params FeedItem[] feed_dict)
{
var results = _run(new object[] { fetches.Item1, fetches.Item2, fetches.Item3, fetches.Item4, fetches.Item5 }, feed_dict);
return (results[0], results[1], results[2], results[3], results[4]);
}

public virtual (NDArray, NDArray, NDArray, NDArray) run((ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation) fetches, params FeedItem[] feed_dict)
{
var results = _run(new object[] {fetches.Item1, fetches.Item2, fetches.Item3, fetches.Item4}, feed_dict);


Loading…
Cancel
Save