|
NEML2 2.0.0
|
Functions | |
| R2 | symmetry (str orbifold, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Crystallography helper routines
| R2 symmetry | ( | str | orbifold, |
| * | , | ||
| torch.dtype | dtype = ..., | ||
| torch.device | device = ..., | ||
| bool | requires_grad = False ) |
Return the symmetry operators for a given symmetry group as a batch of rank two tensors
:param orbifold: String giving the orbifold notation for the symmetry group
:param dtype: Floating point scalar type used throughout the model.
:param device: Device on which the model will be evaluated. All parameters, buffers,
and custom data are synced to the given device.
:param requires_grad: If true, turn on requires_grad in the resulting tensor