def f(c, a):
assert a.shape == (3,)
assert c.shape == (4,)
b = (tf_np.sum(tf_np.sin(a)) + tf_np.sum(tf_np.sin(c)) +
tf_np.sum(tf_np.sin(d)))
c = tf_np.sin(c * b)
assert b.shape == () # pylint: disable=g-explicit-bool-comparison
return c, b
def f(c_g, a_e):
c, g = c_g
a, e = a_e
assert a.shape == (3,)
assert e.shape == () # pylint: disable=g-explicit-bool-comparison
assert c.shape == (4,)
assert g.shape == (2,)
b = tf_np.cos(tf_np.sum(tf_np.sin(a)) + tf_np.sum(tf_np.cos(c)) +
tf_np.sum(tf_np.tan(d)))
f = tf_np.cos(a)
c = tf_np.sin(c * b)
g = tf_np.sin(g * b)
assert b.shape == () # pylint: disable=g-explicit-bool-comparison
assert f.shape == (3,)
return [c, g], (b, f)
def f(c_g_i, a_e_h):
c_g, i = c_g_i
c, g = c_g
a, e_h = a_e_h
e, h = e_h
assert a.shape == (3,)
assert e.shape == () # pylint: disable=g-explicit-bool-comparison
assert c.shape == (4,)
assert g.shape == (2,)
assert i is None
assert h is None
b = tf_np.cos(tf_np.sum(tf_np.sin(a)) + tf_np.sum(tf_np.cos(c)) +
tf_np.sum(tf_np.tan(d)))
f = tf_np.cos(a)
c = tf_np.sin(c * b)
g = tf_np.sin(g * b)
assert b.shape == () # pylint: disable=g-explicit-bool-comparison
assert f.shape == (3,)
return [(c, g), i], (b, [f, h])
