mlx.nn.Module.state

目录

mlx.nn.Module.state#

property Module.state#

模块的状态字典

模块的状态字典包含模块上设置的任何属性,包括 Module.parameters() 中的参数

Module.parameters() 不同,Module.state 属性是对模块状态的引用。对其进行的更新将反映在原始模块中。