助手

目录

助手#

class flax.nnx.Dict(*args, **kwargs)[源代码]#

class flax.nnx.Sequential(*args, **kwargs)[源代码]#

class flax.nnx.TrainState(graphdef: 'GraphDef[M]', params: 'State', opt_state: 'optax.OptState', step: 'jax.Array', tx: 'optax.GradientTransformation')[源代码]#

replace(**updates)#: 返回一个新对象，用新值替换指定的字段。