hookeai.simulators.fetorch.element.derivatives.gradients.vexpand_grad_operator_sym_2d_to_3d

vexpand_grad_operator_sym_2d_to_3d(grad_operator_sym_2d)[source]

Expand 2D discrete symmetric gradient operator to 3D.

Compatible with vectorized mapping.

Parameters:

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

Returns:

grad_operator_sym_3d – Discrete symmetric gradient operator (3D).

Return type:

torch.Tensor(2d)