67118e3d39801c6d970e1f829350e14c5df2cb29,t3f/ops.py,,quadratic_form,#Any#Any#Any#,885

Before Change

    raise ValueError("The arguments should be a TT-matrix.")

  // TODO: make a more efficient implementation taylored for this case.
  return flat_inner(A, tt_tt_matmul(b, transpose(c)))


def cast(tt_a, dtype):

After Change

def quadratic_form(A, b, c):
  if isinstance(b, TensorTrainBatch):
    return quadratic_form_batch(A, b, c)
  if isinstance(b, TensorTrain):
    return quadratic_form_single(A, b, c)


def quadratic_form_batch(A, b, c):
  Computes the quadratic form b^t A c where A is a TT-matrix (or a batch).

Instances