mlx.nn.Module.apply_to_modules

Module.apply_to_modules(apply_fn: Callable[[str, Module], Any]) → Module

对此实例中的所有模块（包括此实例本身）应用一个函数。

参数：

apply_fn (Callable) – 应用于模块的函数。

返回：

更新子模块后的模块实例。