SpatiallyVarying
SpatiallyVarying(value, *, axis=0)Marker for parameters that vary spatially in the domain.
This class wraps a parameter value that has spatial variation along a specified axis, enabling vectorized computations across the domain.
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| value | jax.Array | The parameter values, with shape including the spatial dimension. | required |
| axis | int | The axis along which the parameter varies spatially. Default is 0. | 0 |