@@ -167,7 +167,7 @@ def min(data, axis=None, keepdims=False):
167
167
return cpp .min (data , axis , keepdims )
168
168
169
169
170
- def argmax (data , axis = None , keepdims = False , exclude = False , select_last_index = False ):
170
+ def argmax (data , axis = None , keepdims = False , select_last_index = False ):
171
171
"""Returns the indices of the maximum values along an axis.
172
172
173
173
Parameters
@@ -185,14 +185,18 @@ def argmax(data, axis=None, keepdims=False, exclude=False, select_last_index=Fal
185
185
with size one.
186
186
With this option, the result will broadcast correctly against the input array.
187
187
188
+ select_last_index: bool
189
+ Whether to select the last index if the maximum element appears multiple times, else
190
+ select the first index.
191
+
188
192
Returns
189
193
-------
190
194
ret : tvm.te.Tensor
191
195
"""
192
- return cpp .argmax (data , axis , keepdims , exclude = exclude , select_last_index = select_last_index )
196
+ return cpp .argmax (data , axis , keepdims , select_last_index )
193
197
194
198
195
- def argmin (data , axis = None , keepdims = False , exclude = False , select_last_index = False ):
199
+ def argmin (data , axis = None , keepdims = False , select_last_index = False ):
196
200
"""Returns the indices of the minimum values along an axis.
197
201
198
202
Parameters
@@ -210,11 +214,15 @@ def argmin(data, axis=None, keepdims=False, exclude=False, select_last_index=Fal
210
214
with size one.
211
215
With this option, the result will broadcast correctly against the input array.
212
216
217
+ select_last_index: bool
218
+ Whether to select the last index if the minimum element appears multiple times, else
219
+ select the first index.
220
+
213
221
Returns
214
222
-------
215
223
ret : tvm.te.Tensor
216
224
"""
217
- return cpp .argmin (data , axis , keepdims , exclude , select_last_index )
225
+ return cpp .argmin (data , axis , keepdims , select_last_index )
218
226
219
227
220
228
def prod (data , axis = None , keepdims = False ):
0 commit comments