训练# 实验性 API。 有关更多详细信息,请参阅 NNX 页面。 指标 度量 Metric.__init__() Metric.compute() Metric.reset() Metric.update() 平均 Average.__init__() Average.compute() Average.reset() Average.update() 准确率 Accuracy.update() 韦尔福德 Welford.__init__() Welford.compute() Welford.reset() Welford.update() 多重度量 MultiMetric.__init__() MultiMetric.compute() MultiMetric.reset() MultiMetric.update() 优化器 优化器 Optimizer.step Optimizer.model Optimizer.tx Optimizer.opt_state Optimizer.__init__() Optimizer.update()