apply_to_list¶

pytorch_forecasting.utils.apply_to_list(obj: Union[List[Any], Any], func: Callable) → Union[List[Any], Any][source]¶

Apply function to a list of objects or directly if passed value is not a list.

This is useful if the passed object could be either a list to whose elements a function needs to be applied or just an object to whicht to apply the function.

Parameters

obj (Union[List[Any], Any]) – list/tuple on whose elements to apply function, otherwise object to whom to apply function
func (Callable) – function to apply

Returns

list of objects or object depending on function output and: if input obj is of type list/tuple

Return type

Union[List[Any], Any]

TemporalFusionTransformer

autocorrelation