Ejemplos de sum en Python

Ejemplo n.º 1

 def fun(x):
     return cp.sum(x, axis=(1, 2))

Ejemplo n.º 2

 def fun(x):
     return cp.sum(x, axis=0, keepdims=True)

Ejemplo n.º 3

 def fun(x):
     return cp.sum(x, axis=0)

Ejemplo n.º 4

 def fun(x):
     return cp.sum(x)

Ejemplo n.º 5

def test_nan_to_num():
    y = cp.array([0.0, cp.nan, cp.inf, -cp.inf])
    fun = lambda x: cp.sum(cp.sin(cp.nan_to_num(x + y)))

    x = cp.random.randn(4)
    check_grads(fun)(x)

Ejemplo n.º 6

def test_multi_index2():
    A = cpr.randn(3)
    fun = lambda x: cp.sum(x[[0, 1, 0]])
    check_grads(fun)(A)