应用于列表#

pytorch_forecasting.utils._utils.apply_to_list(obj: List[Any] | Any, func: Callable) List[Any] | Any[来源]#

将函数应用于对象列表,或直接应用于传递的值(如果该值不是列表)。

如果传入的对象可能是一个列表(需要对其元素应用函数),或只是一个需要应用函数的对象,那么这非常有用。

Parameters:
  • obj (Union[List[Any], Any]) – 要应用函数的元素的列表/元组,否则是要应用函数的对象

  • func (Callable) – 要应用的函数

Returns:

对象列表或依赖于函数输出的对象

如果输入 obj 的类型是列表/元组

Return type:

联合[任意类型列表], 任意类型