aesara.tensor.argmax#
- aesara.tensor.argmax(x, axis=None, keepdims=False)[source]#
Returns indices of maximum elements obtained by iterating over given axis.
When axis is None (the default value), the argmax is performed over the flattened tensor.
- Parameters:
keepdims (bool) – If this is set to True, the axes which are reduced are left in the result as dimensions with size one. With this option, the result will broadcast correctly against the original tensor.