fec458e78428cdf828701d5233107aac3664b3cf,t3f/autodiff.py,,hessian_vector_product,#Any#Any#Any#Any#Any#,110

Before Change

      assert_op = tf.no_op()
    with tf.control_dependencies([assert_op]):
      vector_projected = riemannian.project(vector, x)
    cores_grad = tf.gradients(function_value, deltas)
    vec_deltas = riemannian.tangent_space_to_deltas(vector_projected)
    products = [tf.reduce_sum(a * b) for a, b in zip(cores_grad, vec_deltas)]
    grad_times_vec = tf.add_n(products)
    second_cores_grad = tf.gradients(grad_times_vec, deltas)
    final_deltas = _enforce_gauge_conditions(second_cores_grad, left)
    return riemannian.deltas_to_tangent_space(final_deltas, x, left, right)

After Change

      products = [tf.reduce_sum(a * b) for a, b in zip(cores_grad, vec_deltas)]
      return tf.add_n(products)

    _, second_cores_grad = value_and_grad(augmented_outer_func, deltas)
    final_deltas = _enforce_gauge_conditions(second_cores_grad, left)
    return riemannian.deltas_to_tangent_space(final_deltas, x, left, right)

Instances