HyperGP.tensor.argmax

Contents

HyperGP.tensor.argmax#

argmax(x, dim=0)[source]#

Return the indices of the maximum of an array along the dim.

Parameters:

  • x (Tensor or array_like) – Elements to search for maximum values.

  • dim – The dim of x which a ‘max’ is performed.

Returns:

the indices of the maximum of an array along the dim.

Return type:

ret(Tensor)

Examples

import modules

>>> import numpy as np
>>> from HyperGP import Tensor
>>> import time

array initialization

>>> x1 = np.random.uniform(-1, 1, size=(500, 100000))
>>> x1_t = Tensor(x1)

runtime test

>>> st = time.time()
>>> ar = [np.argmax(x1) for i in range(10)]
>>> print("numpy runtime: ", time.time() - st)

>>> st = time.time()
>>> ar = [HyperGP.argmax(x1_t) for i in range(10)]
>>> print("HyperGP runtime: ", time.time() - st)

broadcast operation

>>> ar = [np.argmax(x, axis=1) for i in range(10)]
>>> ar = [HyperGP.argmax(x, dim=1) for i in range(10)]
>>> for x in ar:
...     x.wait()
Contents