desc.objectives.SurfaceCurrentRegularization

class desc.objectives.SurfaceCurrentRegularization(surface_current_field, target=None, bounds=None, weight=1, normalize=True, normalize_target=True, loss_function=None, deriv_mode='auto', source_grid=None, name='surface-current-regularization')Source

Target the surface current magnitude.

compute:

w * ||K|| * sqrt(||e_theta x e_zeta||)

where K is the winding surface current density, w is the regularization parameter (the weight on this objective), and ||e_theta x e_zeta|| is the magnitude of the surface normal i.e. the surface jacobian ||e_theta x e_zeta||

This is intended to be used with a surface current:

K = n x ∇ Φ

i.e. a CurrentPotentialField

Intended to be used with a QuadraticFlux objective, to form a problem similar to the REGCOIL algorithm described in [1] (if used with a FourierCurrentPotentialField, is equivalent to the simple regularization of the solve_regularized_surface_current method).

References

[1] (1,2)

Landreman, Matt. “An improved current potential method for fast computation of stellarator coil shapes.” Nuclear Fusion (2017).

Parameters:

• surface_current_field (CurrentPotentialField) – Surface current which is producing the magnetic field, the parameters of this will be optimized to minimize the objective.

• source_grid (Grid, optional) – Collocation grid containing the nodes to evaluate current source at on the winding surface. If used in conjunction with the QuadraticFlux objective, with its field_grid matching this source_grid, this replicates the REGCOIL algorithm described in [1] .

• target ({float, ndarray}, optional) – Target value(s) of the objective. Only used if bounds is None. Must be broadcastable to Objective.dim_f. Defaults to target=0.

• bounds (tuple of {float, ndarray}, optional) – Lower and upper bounds on the objective. Overrides target. Both bounds must be broadcastable to Objective.dim_f Defaults to target=0.weight : {float, ndarray}, optional Weighting to apply to the Objective, relative to other Objectives. Must be broadcastable to to Objective.dim_f When used with QuadraticFlux objective, this acts as the regularization parameter (with w^2 = lambda), with 0 corresponding to no regularization. The larger this parameter is, the less complex the surface current will be, but the worse the normal field.

• normalize (bool, optional) – Whether to compute the error in physical units or non-dimensionalize.

• normalize_target (bool, optional) – Whether target and bounds should be normalized before comparing to computed values. If normalize is True and the target is in physical units, this should also be set to True.

• loss_function ({None, 'mean', 'min', 'max'}, optional) – Loss function to apply to the objective values once computed. This loss function is called on the raw compute value, before any shifting, scaling, or normalization.

• deriv_mode ({"auto", "fwd", "rev"}) – Specify how to compute Jacobian matrix, either forward mode or reverse mode AD. “auto” selects forward or reverse mode based on the size of the input and output of the objective. Has no effect on self.grad or self.hess which always use reverse mode and forward over reverse mode respectively.

• name (str, optional) – Name of the objective.

• jac_chunk_size (int or "auto", optional) – Will calculate the Jacobian jac_chunk_size columns at a time, instead of all at once. The memory usage of the Jacobian calculation is roughly memory usage = m0 + m1*jac_chunk_size: the smaller the chunk size, the less memory the Jacobian calculation will require (with some baseline memory usage). The time it takes to compute the Jacobian is roughly t= t0 + t1/jac_chunk_size` so the larger the ``jac_chunk_size, the faster the calculation takes, at the cost of requiring more memory. If None, it will use the largest size i.e obj.dim_x. Defaults to chunk_size=None.

Methods

build([use_jit, verbose])	Build constant arrays.
compute([surface_params, constants])	Compute surface current regularization.
compute_scalar(*args, **kwargs)	Compute the scalar form of the objective.
compute_scaled(*args, **kwargs)	Compute and apply weighting and normalization.
compute_scaled_error(*args, **kwargs)	Compute and apply the target/bounds, weighting, and normalization.
compute_unscaled(*args, **kwargs)	Compute the raw value of the objective.
copy([deepcopy])	Return a (deep)copy of this object.
equiv(other)	Compare equivalence between DESC objects.
grad(*args, **kwargs)	Compute gradient vector of self.compute_scalar wrt x.
hess(*args, **kwargs)	Compute Hessian matrix of self.compute_scalar wrt x.
jac_scaled(*args, **kwargs)	Compute Jacobian matrix of self.compute_scaled wrt x.
jac_scaled_error(*args, **kwargs)	Compute Jacobian matrix of self.compute_scaled_error wrt x.
jac_unscaled(*args, **kwargs)	Compute Jacobian matrix of self.compute_unscaled wrt x.
jvp_scaled(v, x[, constants])	Compute Jacobian-vector product of self.compute_scaled.
jvp_scaled_error(v, x[, constants])	Compute Jacobian-vector product of self.compute_scaled_error.
jvp_unscaled(v, x[, constants])	Compute Jacobian-vector product of self.compute_unscaled.
load(load_from[, file_format])	Initialize from file.
print_value(args[, args0])	Print the value of the objective.
save(file_name[, file_format, file_mode])	Save the object.
xs(*things)	Return a tuple of args required by this objective from optimizable things.

Attributes

bounds	Lower and upper bounds of the objective.
built	Whether the transforms have been precomputed (or not).
constants	Constant parameters such as transforms and profiles.
dim_f	Number of objective equations.
fixed	Whether the objective fixes individual parameters (or linear combo).
linear	Whether the objective is a linear function (or nonlinear).
name	Name of objective (str).
normalization	normalizing scale factor.
scalar	Whether default "compute" method is a scalar or vector.
target	Target value(s) of the objective.
things	Optimizable things that this objective is tied to.
weight	Weighting to apply to the Objective, relative to other Objectives.
weight_str