collie.utils.apply_to_collection

apply_to_collection(data, dtype, function, *args, wrong_dtype=None, include_none=True, **kwargs)

递归地对 data 中的元素执行函数 function，且仅在满足元素为 dtype 时执行。

该函数参考了 pytorch-lightning 的实现

参数:

• data (Any) -- 需要进行处理的数据集合或数据；

• dtype (Union[type, Any, Tuple[Union[type, Any]]]) -- 数据的类型，函数 function 只会被应用于 data 中类型为 dtype 的数据；

• function (Callable) -- 对数据进行处理的函数；

• args (Any) -- function 所需要的其它参数；

• wrong_dtype (Union[type, Tuple[type], None], default: None) -- function 一定不会生效的数据类型。 如果数据既是 wrong_dtype 类型又是 dtype 类型那么也不会生效；

• include_none (bool, default: True) -- 是否包含执行结果为 None 的数据，默认为 True；

• kwargs (Any) -- function 所需要的其它参数；

返回类型:

Any

返回:

经过 function 处理后的数据集合；