提交 3841d2c1 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Update docstring after code review

上级 d7d5401c
...@@ -1572,7 +1572,7 @@ def verify_grad(fun, pt, n_tests=2, rng=None, eps=None, ...@@ -1572,7 +1572,7 @@ def verify_grad(fun, pt, n_tests=2, rng=None, eps=None,
and returns a Theano variable. For instance, an Op instance with and returns a Theano variable. For instance, an Op instance with
a single output. a single output.
:param pt: the list of numpy.ndarrays to use as input values. :param pt: the list of numpy.ndarrays to use as input values.
These arrays must be either float32 or float64 arrays. These arrays must be either float16, float32, or float64 arrays.
:param n_tests: number of times to run the test :param n_tests: number of times to run the test
:param rng: random number generator used to sample u, we test gradient :param rng: random number generator used to sample u, we test gradient
of sum(u * fun) at pt of sum(u * fun) at pt
...@@ -1591,7 +1591,7 @@ def verify_grad(fun, pt, n_tests=2, rng=None, eps=None, ...@@ -1591,7 +1591,7 @@ def verify_grad(fun, pt, n_tests=2, rng=None, eps=None,
comparison comparison
:param cast_to_output_type: if the output is float32 and :param cast_to_output_type: if the output is float32 and
cast_to_output_type is True, cast the random projection to cast_to_output_type is True, cast the random projection to
float32. Otherwise it is float64. float32. Otherwise it is float64. float16 is not handled here.
:param no_debug_ref: Don't use DebugMode for the numerical :param no_debug_ref: Don't use DebugMode for the numerical
gradient function. gradient function.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论