From acaf5dfc9467adbb4a82aa00a898f3027eb78370 Mon Sep 17 00:00:00 2001 From: Anton Chernov Date: Wed, 7 Nov 2018 19:20:09 +0100 Subject: [PATCH] add/update infer_range docs (#13153) --- python/mxnet/ndarray/ndarray.py | 3 +++ python/mxnet/symbol/symbol.py | 3 +++ .../core/src/main/scala/org/apache/mxnet/NDArray.scala | 3 +++ 3 files changed, 9 insertions(+) diff --git a/python/mxnet/ndarray/ndarray.py b/python/mxnet/ndarray/ndarray.py index 46b21a90d4c6..c20da0c57314 100644 --- a/python/mxnet/ndarray/ndarray.py +++ b/python/mxnet/ndarray/ndarray.py @@ -2493,6 +2493,9 @@ def arange(start, stop=None, step=1.0, repeat=1, ctx=None, dtype=mx_real_t): Spacing between values. The default step size is 1. repeat : int, optional Number of times to repeat each element. The default repeat count is 1. + infer_range : boolean, optional + When set to True, infer the stop position from the start, step, + repeat, and output tensor size. ctx : Context, optional Device context. Default context is the current default context. dtype : str or numpy.dtype, optional diff --git a/python/mxnet/symbol/symbol.py b/python/mxnet/symbol/symbol.py index 5f6cbd6b6e14..b289cc7ac52b 100644 --- a/python/mxnet/symbol/symbol.py +++ b/python/mxnet/symbol/symbol.py @@ -2900,6 +2900,9 @@ def arange(start, stop=None, step=1.0, repeat=1, name=None, dtype=None): repeat : int, optional "The repeating time of all elements. E.g repeat=3, the element a will be repeated three times --> a, a, a. + infer_range : boolean, optional + When set to True, infer the stop position from the start, step, + repeat, and output tensor size. dtype : str or numpy.dtype, optional The value type of the inner value, default to ``np.float32``. diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala b/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala index 548c30b73a14..033487a59beb 100644 --- a/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala +++ b/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala @@ -403,6 +403,9 @@ object NDArray extends NDArrayBase { * @param stop End of interval. * @param step Spacing between values. The default step size is 1. * @param repeat Number of times to repeat each element. The default repeat count is 1. + * @param infer_range + * When set to True, infer the stop position from the start, step, + * repeat, and output tensor size. * @param ctx Device context. Default context is the current default context. * @param dType The data type of the `NDArray`. The default datatype is `DType.Float32`. * @return NDArray of evenly spaced values in the specified range.