Ejemplo n.º 1
0
def test_deriv_binary_int(func, preserve_result, a, n):
    """Test derivatives of functions with scalar and integer input."""
    utils.test_forward_array(func, (0, ), preserve_result, a, n)
Ejemplo n.º 2
0
def test_deriv_ternary(func, preserve_result, a, b, c):
    """Test derivatives of three-argument scalar functions."""
    utils.test_forward_array(func, (0, ), preserve_result, a, b, c)
Ejemplo n.º 3
0
def test_deriv_unary(func, preserve_result, a):
    """Test derivatives of single-argument scalar functions."""
    utils.test_forward_array(func, (0, ), preserve_result, a)
Ejemplo n.º 4
0
def test_grad_binary_int(func, preserve_result, a, n):
  """Test gradients of functions with scalar and integer input."""
  utils.test_forward_array(func, (0,), preserve_result, a, n)
Ejemplo n.º 5
0
def test_grad_ternary(func, preserve_result, a, b, c):
  """Test gradients of three-argument scalar functions."""
  utils.test_forward_array(func, (0,), preserve_result, a, b, c)
Ejemplo n.º 6
0
def test_grad_unary(func, preserve_result, a):
  """Test gradients of single-argument scalar functions."""
  utils.test_forward_array(func, (0,), preserve_result, a)