mlx.nn.Module.update_modules

Module.update_modules(modules: dict) → Module

用提供的字典（可以是字典或列表的字典）中的子模块替换此 Module 实例的子模块。

它相当于 Module.update() 方法，但作用于模块而非参数，使我们能够通过编程方式灵活地编辑复杂架构，实现层的替换。

传入的模块字典不必像 parameters() 方法那样是完整的字典。只有提供的位置会被更新。

参数:

modules (dict) – 模块子模块的完整或部分字典。

返回:

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