训练# 实验性 API。有关更多详细信息,请参阅 NNX 页面。 指标 指标 Metric.__init__() Metric.compute() Metric.reset() Metric.update() 平均值 Average.__init__() Average.compute() Average.reset() Average.update() 准确率 Accuracy.update() Welford 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()