訓練# 實驗性 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 MultiMetric.__init__() MultiMetric.compute() MultiMetric.reset() MultiMetric.update() 最佳化器 最佳化器 Optimizer.step Optimizer.model Optimizer.tx Optimizer.opt_state Optimizer.__init__() Optimizer.update()