4.1.1.1.2 pde.backends.jax.operators.cylindrical_sym module

This module implements differential operators on cylindrical grids.

make_laplace

Make a discretized laplace operator for a cylindrical grid.

make_gradient

Make a discretized gradient operator for a cylindrical grid.

make_gradient_squared

Make a discretized gradient squared operator for a cylindrical grid.

make_divergence

Make a discretized divergence operator for a cylindrical grid.

make_vector_gradient

Make a discretized vector gradient operator for a cylindrical grid.

make_vector_laplace

Make a discretized vector laplace operator for a cylindrical grid.

make_tensor_divergence

Make a discretized tensor divergence operator for a cylindrical grid.

make_divergence(grid)[source]

Make a discretized divergence operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_gradient(grid)[source]

Make a discretized gradient operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_gradient_squared(grid, *, central=True)[source]

Make a discretized gradient squared operator for a cylindrical grid.

Parameters:
  • grid (CylindricalSymGrid) – The grid for which the operator is created

  • central (bool) – Whether a central difference approximation is used for the gradient operator. If this is False, the squared gradient is calculated as the mean of the squared values of the forward and backward derivatives.

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_laplace(grid)[source]

Make a discretized laplace operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_tensor_divergence(grid)[source]

Make a discretized tensor divergence operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_vector_gradient(grid)[source]

Make a discretized vector gradient operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType

make_vector_laplace(grid)[source]

Make a discretized vector laplace operator for a cylindrical grid.

Parameters:

grid (CylindricalSymGrid) – The grid for which the operator is created

Returns:

A function that can be applied to an array of values

Return type:

OperatorImplType