示例#1
0
def test_grad_refactor_simple_1():
    assert C.grad_all(grad_refactor_simple_1)(2, 1) == (4, 2)
示例#2
0
def test_grad_refactor_10():
    """ test_grad_while """
    assert C.grad_all(grad_refactor__while_1)(Tensor(5,
                                                     dtype=ms.int32)) == (60, )
示例#3
0
def first_derivative_all(x):
    """ first_derivative_all """
    return grad_all(single)(x)[0]
示例#4
0
def test_grad_refactor_4():
    assert C.grad_all(grad_refactor_4)(Tensor(4, dtype=ms.int32)) == (3, )
示例#5
0
def test_grad_refactor_6():
    assert C.grad_all(grad_refactor_6)(Tensor(3, dtype=ms.int32),
                                       Tensor(2, dtype=ms.int32)) == (3, 1)
示例#6
0
def test_grad_refactor_simple_1():
    assert C.grad_all(grad_refactor_simple_1)(Tensor(2, dtype=ms.int32),
                                              Tensor(1,
                                                     dtype=ms.int32)) == (4, 2)
示例#7
0
def test_grad_refactor_2():
    assert C.grad_all(grad_refactor_2)(Tensor(2, dtype=ms.int32),
                                       Tensor(3, dtype=ms.int32)) == (27, 54)
示例#8
0
def test_grad_refactor_5():
    assert C.grad_all(grad_refactor_5)(1) == (1, )
示例#9
0
def test_grad_refactor_6():
    assert C.grad_all(grad_refactor_6)(3, 2) == (3, 1)
示例#10
0
def test_grad_refactor_3():
    assert C.grad_all(grad_refactor_3)(3) == (3, )
示例#11
0
def test_grad_refactor_4():
    assert C.grad_all(grad_refactor_4)(4) == (3, )
示例#12
0
def test_grad_refactor_2():
    assert C.grad_all(grad_refactor_2)(2, 3) == (27, 54)
示例#13
0
def test_grad_refactor_1():
    assert C.grad_all(grad_refactor_1)(2, 3) == (3, 2)
示例#14
0
def test_grad_refactor_simple_2():
    assert C.grad_all(grad_refactor_simple_2)(2, 3, 0) == (7, 4, 7)
示例#15
0
def grad_while(x):
    """ grad_while """
    return C.grad_all(_while)(x)
示例#16
0
def test_grad_refactor_9():
    assert C.grad_all(grad_refactor_while)(3) == (6, )
示例#17
0
def grad_for(x):
    """ grad_for """
    return C.grad_all(_for)(x)
示例#18
0
def test_grad_refactor_10():
    """ test_grad_while """
    assert C.grad_all(grad_refactor__while_1)(5) == (60, )
示例#19
0
def test_grad_refactor_simple_2():
    x = Tensor(2, dtype=ms.int32)
    y = Tensor(3, dtype=ms.int32)
    z = Tensor(0, dtype=ms.int32)
    assert C.grad_all(grad_refactor_simple_2)(x, y, z) == (7, 4, 7)
示例#20
0
def test_grad_refactor_14():
    assert C.grad_all(grad_refactor_14)(2, 3) == (3, 9)
示例#21
0
 def df_refactor_3(x):
     return C.grad_all(grad_refactor_3)(x)
示例#22
0
 def df_refactor_14(x, y):
     return C.grad_all(grad_refactor_14)(x, y)
示例#23
0
 def df_refactor_5(x):
     return C.grad_all(grad_refactor_5)(x)
示例#24
0
def mainf(x, y):
    """ mainf """
    return C.grad_all(mul)(x, y)
示例#25
0
 def df_refactor_while(input_x):
     return C.grad_all(grad_refactor_while)(input_x)
示例#26
0
def grad_add_mul(x, y):
    """ grad_add_mul """
    return C.grad_all(add_mul)(x, y)
示例#27
0
 def construct(self, x, y, b):
     return C.grad_all(self.network)(x, y, b)
示例#28
0
def grad_if(x, y):
    """ grad_if """
    return C.grad_all(if_test)(x, y)
示例#29
0
def second_derivative_all(x):
    """ second_derivative_all """
    return grad_all(first_derivative_all)(x)[0]
示例#30
0
 def f(x, y):
     return C.grad_all(debug_test)(x, y)