Sweep

Inheritance Diagram

Inheritance diagram of tenpy.algorithms.mps_common.Sweep

Methods

Sweep.__init__(psi, model, options)

Initialize self.

Sweep.environment_sweeps(N_sweeps)

Perform N_sweeps sweeps without optimization to update the environment.

Sweep.get_sweep_schedule()

Define the schedule of the sweep.

Sweep.init_env([model])

(Re-)initialize the environment.

Sweep.make_eff_H()

Create new instance of self.EffectiveH at self.i0 and set it to self.eff_H.

Sweep.post_update_local(update_data)

Algorithm-specific actions to be taken after local update.

Sweep.prepare_update()

Prepare everything algorithm-specific to perform a local update.

Sweep.reset_stats()

Reset the statistics.

Sweep.sweep([optimize])

One ‘sweep’ of a sweeper algorithm.

Sweep.update_LP(_)

Sweep.update_RP(_)

Sweep.update_local(theta, **kwargs)

Perform algorithm-specific local update.

Class Attributes and Properties

Sweep.engine_params

class tenpy.algorithms.mps_common.Sweep(psi, model, options)[source]

Bases: object

Prototype class for a ‘sweeping’ algorithm.

This is a superclass, intended to cover common procedures in all algorithms that ‘sweep’. This includes DMRG, TDVP, etc. Only DMRG is currently implemented in this way.

Parameters
  • psi (MPS) – Initial guess for the ground state, which is to be optimized in-place.

  • model (MPOModel) – The model representing the Hamiltonian for which we want to find the ground state.

  • options (dict) – Further optional configuration parameters.

Options

config Sweep
option summary

combine

Whether to combine legs into pipes. This combines the virtual and [...]

init_env_data in DMRGEngine.init_env

Dictionary as returned by ``self.env.get_initialization_data()`` from [...]

lanczos_params

Lanczos parameters as described in [...]

orthogonal_to in DMRGEngine.init_env

List of other matrix product states to orthogonalize against. [...]

start_env in DMRGEngine.init_env

Number of sweeps to be performed without optimization to update [...]

sweep_0 in Sweep.reset_stats

Number of sweeps that have already been performed.

trunc_params

Truncation parameters as described in :cfg:config:`truncation`.

verbose

Level of verbosity (i.e. how much status information to print); higher=more [...]

option combine: bool

Whether to combine legs into pipes. This combines the virtual and physical leg for the left site (when moving right) or right side (when moving left) into pipes. This reduces the overhead of calculating charge combinations in the contractions, but one matvec() is formally more expensive, \(O(2 d^3 \chi^3 D)\).

option lanczos_params: Config

Lanczos parameters as described in lanczos()

option trunc_params: dict

Truncation parameters as described in truncation.

option verbose: bool | int

Level of verbosity (i.e. how much status information to print); higher=more output.

options

Optional parameters.

Type

Config

E_trunc_list

List of truncation energies throughout a sweep.

Type

list

env

Environment for contraction <psi|H|psi>.

Type

MPOEnvironment

finite

Whether the MPS boundary conditions are finite (True) or infinite (False)

Type

bool

i0

Only set during sweep. Left-most of the EffectiveH.length sites to be updated in update_local().

Type

int

move_right

Only set during sweep. Whether the next i0 of the sweep will be right or left of the current one.

Type

bool

ortho_to_envs

List of environments <psi|psi_ortho>, where psi_ortho is an MPS to orthogonalize against.

Type

list of MPSEnvironment

shelve

If a simulation runs out of time (time.time() - start_time > max_seconds), the run will terminate with shelve = True.

Type

bool

sweeps

The number of sweeps already performed. (Useful for re-start).

Type

int

time0

Time marker for the start of the run.

Type

float

trunc_err_list

List of truncation errors.

Type

list

update_LP_RP

Only set during a sweep. Whether it is necessary to update the LP and RP. The latter are chosen such that the environment is growing for infinite systems, but we only keep the minimal number of environment tensors in memory (inside env).

Type

(bool, bool)

init_env(model=None)[source]

(Re-)initialize the environment.

This function is useful to (re-)start a Sweep with a slightly different model or different (engine) parameters. Note that we assume that we still have the same psi. Calls reset_stats().

Parameters

model (MPOModel) – The model representing the Hamiltonian for which we want to find the ground state. If None, keep the model used before.

Options

Deprecated since version 0.6.0: Options LP, LP_age, RP and RP_age are now collected in a dictionary init_env_data with different keys init_LP, init_RP, age_LP, age_RP

option Sweep.init_env_data: dict

Dictionary as returned by self.env.get_initialization_data() from get_initialization_data().

option Sweep.orthogonal_to: list of MPSEnvironment

List of other matrix product states to orthogonalize against. Works only for finite systems. This parameter can be used to find (a few) excited states as follows. First, run DMRG to find the ground state and then run DMRG again while orthogonalizing against the ground state, which yields the first excited state (in the same symmetry sector), and so on.

option Sweep.start_env: int

Number of sweeps to be performed without optimization to update the environment.

Raises

ValueError – If the engine is re-initialized with a new model, which legs are incompatible with those of hte old model.

reset_stats()[source]

Reset the statistics. Useful if you want to start a new Sweep run.

This method is expected to be overwritten by subclass, and should then define self.update_stats and self.sweep_stats dicts consistent with the statistics generated by the algorithm particular to that subclass.

option Sweep.sweep_0: int

Number of sweeps that have already been performed.

environment_sweeps(N_sweeps)[source]

Perform N_sweeps sweeps without optimization to update the environment.

Parameters

N_sweeps (int) – Number of sweeps to run without optimization

sweep(optimize=True)[source]

One ‘sweep’ of a sweeper algorithm.

Iteratate over the bond which is optimized, to the right and then back to the left to the starting point. If optimize=False, don’t actually diagonalize the effective hamiltonian, but only update the environment.

Parameters

optimize (bool, optional) – Whether we actually optimize to find the ground state of the effective Hamiltonian. (If False, just update the environments).

Returns

max_trunc_err – Maximal truncation error introduced.

Return type

float

get_sweep_schedule()[source]

Define the schedule of the sweep.

One ‘sweep’ is a full sequence from the leftmost site to the right and back. Only those LP and RP that can be used later should be updated.

Returns

schedule – Schedule for the sweep. Each entry is (i0, move_right, (update_LP, update_RP)), where i0 is the leftmost of the self.EffectiveH.length sites to be updated in update_local(), move_right indicates whether the next i0 in the schedule is rigth (True) of the current one, and update_LP, update_RP indicate whether it is necessary to update the LP and RP. The latter are chosen such that the environment is growing for infinite systems, but we only keep the minimal number of environment tensors in memory.

Return type

iterable of (int, bool, (bool, bool))

prepare_update()[source]

Prepare everything algorithm-specific to perform a local update.

update_local(theta, **kwargs)[source]

Perform algorithm-specific local update.

post_update_local(update_data)[source]

Algorithm-specific actions to be taken after local update.

An example would be to collect statistics.

make_eff_H()[source]

Create new instance of self.EffectiveH at self.i0 and set it to self.eff_H.