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

Before Change

    right = decompositions.orthogonalize_tt_cores(left, left_to_right=False)
    deltas = [right.tt_cores[0]]
    deltas += [tf.zeros_like(cc) for cc in right.tt_cores[1:]]
    x_projection = riemannian.deltas_to_tangent_space(deltas, x, left, right)
    function_value = func(x_projection)
    if runtime_check:
      assert_op = _is_invariant_to_input_transforms(function_value, func(x))
    else:
      assert_op = tf.no_op()

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