CellOptimState

class torch_sim.optimizers.cell_filters.CellOptimState(positions, masses, cell, pbc, atomic_numbers, system_idx, *, forces, energy, stress, reference_cell, cell_filter, cell_factor=<factory>, pressure=<factory>, hydrostatic_strain=False, constant_volume=False, cell_positions=<factory>, cell_forces=<factory>, cell_masses=<factory>)

Bases: OptimState

State class for cell optimization.

Parameters:

• positions (Tensor)

• masses (Tensor)

• cell (Tensor)

• pbc (bool)

• atomic_numbers (Tensor)

• system_idx (Tensor)

• forces (Tensor)

• energy (Tensor)

• stress (Tensor)

• reference_cell (Tensor)

• cell_filter (tuple[Callable[[...], None], Callable[[...], None]])

• cell_factor (Tensor)

• pressure (Tensor)

• hydrostatic_strain (bool)

• constant_volume (bool)

• cell_positions (Tensor)

• cell_forces (Tensor)

• cell_masses (Tensor)