Python _gen_x0の例

コード例 #1

 def pickeig(w, v, nr, envs):
     x0 = linalg_helper._gen_x0(envs['v'], envs['xs'])
     idx = np.argmax(np.abs(
         np.dot(np.array(guess).conj(),
                np.array(x0).T)),
                     axis=1)
     return lib.linalg_helper._eigs_cmplx2real(w, v, idx)

コード例 #2

ファイル: rccsd_slow.py プロジェクト: xlzan/pyscf

 def pickeig(w, v, nr, envs):
     x0 = linalg_helper._gen_x0(envs['v'], envs['xs'])
     idx = np.argmax(np.abs(
         np.dot(np.array(guess).conj(),
                np.array(x0).T)),
                     axis=1)
     return w[idx].real, v[:, idx].real, idx

コード例 #3

 def pickeig(w, v, nroots, envs):
     x0 = linalg_helper._gen_x0(envs['v'], envs['xs'])
     s = np.dot(np.asarray(guess_k).conj(), np.asarray(x0).T)
     snorm = np.einsum('pi,pi->i', s.conj(), s)
     idx = np.argsort(-snorm)[:nroots]
     return linalg_helper._eigs_cmplx2real(w,
                                           v,
                                           idx,
                                           real_eigenvectors=False)

コード例 #4

ファイル: rccsd.py プロジェクト: eronca/pyscf

 def pickeig(w, v, nr, envs):
     x0 = linalg_helper._gen_x0(envs['v'], envs['xs'])
     idx = np.argmax( np.abs(np.dot(np.array(guess).conj(),np.array(x0).T)), axis=1 )
     return w[idx].real, v[:,idx].real, idx

コード例 #5

ファイル: rccsd_slow.py プロジェクト: chrinide/pyscf

 def pickeig(w, v, nr, envs):
     x0 = linalg_helper._gen_x0(envs['v'], envs['xs'])
     idx = np.argmax( np.abs(np.dot(np.array(guess).conj(),np.array(x0).T)), axis=1 )
     return lib.linalg_helper._eigs_cmplx2real(w, v, idx)