This is an automated email from the ASF dual-hosted git repository.
nswamy pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 76d5197 add/update infer_range docs (#12879)
76d5197 is described below
commit 76d5197b4224ba42f0a932d9ca7b1c8ab80527e7
Author: Aaron Markham <[email protected]>
AuthorDate: Fri Oct 19 17:43:41 2018 -0700
add/update infer_range docs (#12879)
---
python/mxnet/ndarray/ndarray.py | 3 +++
python/mxnet/symbol/symbol.py | 3 +++
scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala | 3 +++
scala-package/core/src/main/scala/org/apache/mxnet/Symbol.scala | 4 +++-
src/operator/tensor/init_op.h | 4 ++--
5 files changed, 14 insertions(+), 3 deletions(-)
diff --git a/python/mxnet/ndarray/ndarray.py b/python/mxnet/ndarray/ndarray.py
index c592465..bf1140d 100644
--- a/python/mxnet/ndarray/ndarray.py
+++ b/python/mxnet/ndarray/ndarray.py
@@ -2547,6 +2547,9 @@ def arange(start, stop=None, step=1.0, repeat=1,
infer_range=False, ctx=None, dt
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 eaf22f3..c657507 100644
--- a/python/mxnet/symbol/symbol.py
+++ b/python/mxnet/symbol/symbol.py
@@ -2926,6 +2926,9 @@ def arange(start, stop=None, step=1.0, repeat=1,
infer_range=False, name=None, d
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 f2a7603..f9f2dbe 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.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/Symbol.scala
b/scala-package/core/src/main/scala/org/apache/mxnet/Symbol.scala
index a009e7e..4472a84 100644
--- a/scala-package/core/src/main/scala/org/apache/mxnet/Symbol.scala
+++ b/scala-package/core/src/main/scala/org/apache/mxnet/Symbol.scala
@@ -974,7 +974,9 @@ object Symbol extends SymbolBase {
* @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 Infer the stop value from output shape
+ * @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.
diff --git a/src/operator/tensor/init_op.h b/src/operator/tensor/init_op.h
index 1a4790a..4e52b08 100644
--- a/src/operator/tensor/init_op.h
+++ b/src/operator/tensor/init_op.h
@@ -161,8 +161,8 @@ struct RangeParam : public dmlc::Parameter<RangeParam> {
" E.g repeat=3, the element a will be repeated three times -->
a, a, a.");
DMLC_DECLARE_FIELD(infer_range)
.set_default(false)
- .describe("Whether to infer the stop position from the start, step,
repeat, and output tensor"
- "size.");
+ .describe("When set to True, infer the stop position from the start, step,
"
+ "repeat, and output tensor size.");
DMLC_DECLARE_FIELD(ctx)
.set_default("")
.describe("Context of output, in format [cpu|gpu|cpu_pinned](n)."
