HyperGP.tensor.sum

Contents

HyperGP.tensor.sum#

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

Sum of array elements along with the corresponding dim.

Parameters:

  • x (Tensor or array_like) – Elements to sum.

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

Returns:

a new ‘Tensor’ is returned

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.sum(x) for i in range(10)]
>>> print("numpy runtime: ", time.time() - st)

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

broadcast operation

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