mlx.nn.Module.update

Module.update(parameters: dict) → Module

用提供的字典和列表中的参数替换此模块的参数。

通常由优化器用于将模型更改为更新后的（优化过的）参数。也由 mlx.nn.value_and_grad() 用于在模型中设置跟踪器以计算梯度。

传入的参数字典不必是类似于 parameters() 的完整字典。只会更新提供的部分。

参数:

parameters (dict) – 模块参数的完整或部分字典。

返回:

更新参数后的模块实例。