Source code for pysisyphus.diabatization.helpers

import numpy as np
from scipy.stats import special_ortho_group


[docs] def get_random_U(N): """Get random rotation matrix.""" return special_ortho_group.rvs(N)
[docs] def fmt_tensor(tensor): return np.array2string(tensor, formatter={"float": lambda f: f"{f: >10.4e}"})