diff --git a/quantarhei/qm/propagators/oqssvevolution.py b/quantarhei/qm/propagators/oqssvevolution.py index bad4ef8..aacd08f 100644 --- a/quantarhei/qm/propagators/oqssvevolution.py +++ b/quantarhei/qm/propagators/oqssvevolution.py @@ -34,6 +34,13 @@ def set_initial_condition(self, psii): self.data[0,:] = psii.data + def get_norm(self): + """Time dependent norm of the state vector + + """ + return numpy.sum(self.data*self.data, axis=1) + + def get_ReducedDensityMatrixEvolution(self, decoherence=False): """ Returns the corresponding reduced density matrix