Skip to content

Commit 1d9e2e5

Browse files
committed
overload session.run for 4 tuples.
1 parent 387ae4c commit 1d9e2e5

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/TensorFlowNET.Core/Sessions/BaseSession.cs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,14 @@ public virtual NDArray run(ITensorOrOperation fetche, params FeedItem[] feed_dic
6868
return _run(fetche, feed_dict)[0];
6969
}
7070

71+
public virtual (NDArray, NDArray, NDArray, NDArray, NDArray) run(
72+
(ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation) fetches,
73+
params FeedItem[] feed_dict)
74+
{
75+
var results = _run(new object[] { fetches.Item1, fetches.Item2, fetches.Item3, fetches.Item4, fetches.Item5 }, feed_dict);
76+
return (results[0], results[1], results[2], results[3], results[4]);
77+
}
78+
7179
public virtual (NDArray, NDArray, NDArray, NDArray) run((ITensorOrOperation, ITensorOrOperation, ITensorOrOperation, ITensorOrOperation) fetches, params FeedItem[] feed_dict)
7280
{
7381
var results = _run(new object[] {fetches.Item1, fetches.Item2, fetches.Item3, fetches.Item4}, feed_dict);

0 commit comments

Comments
 (0)