Commit 01f52b1d by Siju Samuel Committed by Tianqi Chen

[Documentation Changes] Parameter description change for reduction apis (#881)

* [Documentation Changes] Parameter description change for reduction api

The parameter description is updated for max, min, argmax and argmin

* Lint changes on patch-3 of reduction.py
parent 523b6a6b
...@@ -186,9 +186,9 @@ def max(data, axis=None, keepdims=False): ...@@ -186,9 +186,9 @@ def max(data, axis=None, keepdims=False):
The input tvm tensor The input tvm tensor
axis : None or int or tuple of int axis : None or int or tuple of int
Axis or axes along which a sum is performed. Axis or axes along which the max operation is performed.
The default, axis=None, will sum all of the elements of the input array. The default, axis=None, will find the max element from all of the elements of the input
If axis is negative it counts from the last to the first axis. array. If axis is negative it counts from the last to the first axis.
keepdims : bool keepdims : bool
If this is set to True, the axes which are reduced are left in the result as dimensions If this is set to True, the axes which are reduced are left in the result as dimensions
...@@ -212,9 +212,9 @@ def min(data, axis=None, keepdims=False): ...@@ -212,9 +212,9 @@ def min(data, axis=None, keepdims=False):
The input tvm tensor The input tvm tensor
axis : None or int or tuple of int axis : None or int or tuple of int
Axis or axes along which a sum is performed. Axis or axes along which a minimum operation is performed.
The default, axis=None, will sum all of the elements of the input array. The default, axis=None, will find the minimum element from all of the elements of the
If axis is negative it counts from the last to the first axis. input array. If axis is negative it counts from the last to the first axis.
keepdims : bool keepdims : bool
If this is set to True, the axes which are reduced are left in the result as dimensions If this is set to True, the axes which are reduced are left in the result as dimensions
...@@ -238,9 +238,9 @@ def argmax(data, axis=None, keepdims=False): ...@@ -238,9 +238,9 @@ def argmax(data, axis=None, keepdims=False):
The input tvm tensor The input tvm tensor
axis : None or int or tuple of int axis : None or int or tuple of int
Axis or axes along which a sum is performed. Axis or axes along which a argmax operation is performed.
The default, axis=None, will sum all of the elements of the input array. The default, axis=None, will find the indices of the maximum element of the elements of
If axis is negative it counts from the last to the first axis. the input array. If axis is negative it counts from the last to the first axis.
keepdims : bool keepdims : bool
If this is set to True, the axes which are reduced are left in the result as dimensions If this is set to True, the axes which are reduced are left in the result as dimensions
...@@ -265,9 +265,9 @@ def argmin(data, axis=None, keepdims=False): ...@@ -265,9 +265,9 @@ def argmin(data, axis=None, keepdims=False):
The input tvm tensor The input tvm tensor
axis : None or int or tuple of int axis : None or int or tuple of int
Axis or axes along which a sum is performed. Axis or axes along which a argmin operation is performed.
The default, axis=None, will sum all of the elements of the input array. The default, axis=None, will find the indices of minimum element all of the elements of
If axis is negative it counts from the last to the first axis. the input array. If axis is negative it counts from the last to the first axis.
keepdims : bool keepdims : bool
If this is set to True, the axes which are reduced are left in the result as dimensions If this is set to True, the axes which are reduced are left in the result as dimensions
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment