hookeai.simulators.fetorch.element.derivatives.gradients.vreduce_grad_operator_sym_3d_to_2d

vreduce_grad_operator_sym_3d_to_2d(grad_operator_sym_3d)[source]

Reduce 3D discrete symmetric gradient operator to 2D.

Compatible with vectorized mapping.

Parameters:

grad_operator_sym_3d (torch.Tensor(2d)) – Discrete symmetric gradient operator (3D).

Returns:

grad_operator_sym_2d – Discrete symmetric gradient operator (2D).

Return type:

torch.Tensor(2d)