mattertune.backbones.util

Functions

voigt_6_to_full_3x3_stress_torch(stress_vector)

Form a 3x3 stress matrix from a 6 component vector in Voigt notation

mattertune.backbones.util.voigt_6_to_full_3x3_stress_torch(stress_vector)

Form a 3x3 stress matrix from a 6 component vector in Voigt notation

Parameters:

stress_vector (Tensor) – Tensor of shape (B, 6) where B is the batch size

Returns:

Tensor of shape (B, 3, 3)

Return type:

Tensor