From c59a10947d6e9c05c060c8cf96f188c3c517bc2a Mon Sep 17 00:00:00 2001 From: JohannesAck Date: Wed, 14 Nov 2018 06:03:48 +0900 Subject: [PATCH] Parameter documentation for tf_util.function (#349) * Added parameter documentation This parameter was thus far not documented and is non-intuitive when unfamiliar with tf. * Added parameter documentation --- baselines/common/tf_util.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/baselines/common/tf_util.py b/baselines/common/tf_util.py index 2699035..452871c 100644 --- a/baselines/common/tf_util.py +++ b/baselines/common/tf_util.py @@ -165,6 +165,10 @@ def function(inputs, outputs, updates=None, givens=None): outputs: [tf.Variable] or tf.Variable list of outputs or a single output to be returned from function. Returned value will also have the same shape. + updates: [tf.Operation] or tf.Operation + list of update functions or single update function that will be run whenever + the function is called. The return is ignored. + """ if isinstance(outputs, list): return _Function(inputs, outputs, updates, givens=givens)