D | lanczos.py | 120 def gram_schmidt_step(j, basis, v): argument 123 basis_vec = read_colvec(basis, j) 126 return j + 1, basis, v 128 def orthogonalize_once(i, basis, v): argument 130 _, _, v = control_flow_ops.while_loop(lambda j, basis, v: j < i, 131 gram_schmidt_step, [j, basis, v]) 137 def orthogonalize_(i, basis, v): argument 139 v_new, v_new_norm = orthogonalize_once(i, basis, v) 145 lambda: orthogonalize_once(i, basis, v),
|