hookeai.model_architectures.materials.stress_features.build_stress_from_comps

build_stress_from_comps(n_dim, stress_comps_order, stress_comps_array, device=None)

Build stress tensor from given components.

Parameters:

• n_dim (int) – Number of spatial dimensions.

• stress_comps_order (tuple[str]) – Stress components order.

• stress_comps_array (torch.Tensor(1d)) – Stress components array.

• device (torch.device, default=None) – Device on which torch.Tensor is allocated.

Returns:

stress – Stress tensor.

Return type:

torch.Tensor(2d)