Exemple #1
0
def test_scalar_cast_grad():
    """ test_scalar_cast_grad """
    input_x = 255.5
    input_t = get_py_obj_dtype(ms.int8)

    def fx_cast(x):
        output = F.scalar_cast(x, input_t)
        return output

    gfn = C.grad(fx_cast)(input_x)
    expect_dx = 1
    assert gfn == expect_dx
Exemple #2
0
def first_derivative_if(x):
    """ first_derivative_if """
    return grad(if_test)(x)
Exemple #3
0
def third_derivative(x):
    """ third_derivative """
    return grad(second_derivative)(x)
Exemple #4
0
def second_derivative(x):
    """ second_derivative """
    return grad(first_derivative)(x)
Exemple #5
0
def first_derivative(x):
    """ first_derivative """
    return grad(single)(x)
Exemple #6
0
def second_derivative_if(x):
    """ second_derivative_if """
    return grad(first_derivative_if)(x)
Exemple #7
0
 def grad_fx_cast(input_x):
     return C.grad(fx_cast)(input_x)
Exemple #8
0
 def invoke_while(x, y):
     return C.grad(dont_unroll_while)(x, y)
Exemple #9
0
 def df_factorial(x):
     return C.grad(factorial)(x)
Exemple #10
0
def grad_rec(input_x):
    return C.grad(rec)(input_x)
Exemple #11
0
 def df_t2_while(input_x, input_y):
     return C.grad(t2_while)(input_x, input_y)
Exemple #12
0
def test_grad_factorial():
    res = C.grad(factorial)(3)
    assert res == 11
Exemple #13
0
def test_grad_while2():
    res = C.grad(t2_while)(2, 3)
    assert res == 3
Exemple #14
0
def test_grad_rec():
    """ test_grad_rec """
    res = C.grad(rec)(10)
    assert res == 1
Exemple #15
0
 def construct(self, x, y):
     return C.grad(self.network)(x, y)
Exemple #16
0
def mainf(x, y):
    return C.grad(mul)(x, y)
Exemple #17
0
 def construct(self, input_):
     return C.grad(self.network)(input_)
Exemple #18
0
def force_fn(R: Tensor):
    return -C.grad(energy_fn)(R)