tfp.experimental.fastgp.partial_lanczos.gram_schmidt

Return a vector after applying the Gram-Schmidt process to it.

Q A tensor of shape (batch_dims, m, n) representing a batch of orthonormal m-by-n matrices. The m vectors to perform Gram-Schmidt with respect to.
v Initial starting vector of shape (n, batch_dims).

A tensor t of shape (n, batch_dims) such that for every i, t[:, i] is orthogonal to Q[i, j, :] for every j.