hookeai.model_architectures.materials.stress_features.store_stress_comps

store_stress_comps(stress_comps_order, stress, device=None)[source]

Store stress tensor components in array.

Parameters:

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

  • stress (torch.Tensor(2d)) – Stress tensor.

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

Returns:

stress_comps_array – Stress components array.

Return type:

torch.Tensor(1d)