Utility Functions

Utility functions for components of higher.

higher.utils.flatten(x)

Returns a flattened list of objects from a nested structure.

Return type

List[Any]

higher.utils.get_func_params(module, device=None, safe_copy=True)

Returns a detached copy of module parameters which requires gradient.

Return type

List[Tensor]