org.nd4j.linalg.factory.Nd4j.averageAndPropagate()方法的使用及代码示例

x33g5p2x  于2022-01-24 转载在 其他  
字(1.8k)|赞(0)|评价(0)|浏览(113)

本文整理了Java中org.nd4j.linalg.factory.Nd4j.averageAndPropagate()方法的一些代码示例,展示了Nd4j.averageAndPropagate()的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。Nd4j.averageAndPropagate()方法的具体详情如下:
包路径:org.nd4j.linalg.factory.Nd4j
类名称:Nd4j
方法名:averageAndPropagate

Nd4j.averageAndPropagate介绍

[英]This method averages input arrays, and returns averaged array. On top of that, averaged array is propagated to all input arrays
[中]此方法平均输入数组,并返回平均数组。除此之外,平均数组被传播到所有输入数组

代码示例

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper_2.11

  1. private double getScore(AtomicInteger locker) {
  2. wasAveraged = true;
  3. double score = 0.0;
  4. List<INDArray> params = new ArrayList<>();
  5. for (int cnt = 0; cnt < workers && cnt < locker.get(); cnt++) {
  6. params.add(zoo[cnt].getModel().params());
  7. score += zoo[cnt].getModel().score();
  8. }
  9. Nd4j.averageAndPropagate(null, params);
  10. score /= Math.min(workers, locker.get());
  11. // TODO: improve this
  12. if (reportScore)
  13. log.info("Averaged score: " + score);
  14. return score;
  15. }

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper

  1. private double getScore(AtomicInteger locker) {
  2. wasAveraged = true;
  3. double score = 0.0;
  4. List<INDArray> params = new ArrayList<>();
  5. for (int cnt = 0; cnt < workers && cnt < locker.get(); cnt++) {
  6. params.add(zoo[cnt].getModel().params());
  7. score += zoo[cnt].getModel().score();
  8. }
  9. Nd4j.averageAndPropagate(null, params);
  10. score /= Math.min(workers, locker.get());
  11. // TODO: improve this
  12. if (reportScore)
  13. log.info("Averaged score: " + score);
  14. return score;
  15. }

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper

  1. Nd4j.averageAndPropagate(null, updaters);
  2. batchSize += workerModel.batchSize();
  3. Nd4j.averageAndPropagate(null, updaters);

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper_2.11

  1. Nd4j.averageAndPropagate(null, updaters);
  2. batchSize += workerModel.batchSize();
  3. Nd4j.averageAndPropagate(null, updaters);

相关文章