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

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

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

Nd4j.fromByteArray介绍

[英]Read an ndarray from a byte array
[中]

代码示例

代码示例来源:origin: de.datexis/texoo-core

  1. /**
  2. * Get the Vector/Embedding added to this Span. If no Vector was added, return null.
  3. * @param identifier The identifier for this vector.
  4. * @return A previously added INDArray or null
  5. */
  6. public INDArray getVector(String identifier) {
  7. if(vectors != null && vectors.containsKey(identifier)) {
  8. try {
  9. final byte[] vec = vectors.get(identifier);
  10. return Nd4j.fromByteArray(vec);
  11. } catch(IOException ex) {
  12. log.error("IOError in putVector(): {}", ex.toString());
  13. return null;
  14. }
  15. } else {
  16. log.error("Requesting unknown vector with identifier '" + identifier + "'");
  17. return null;
  18. }
  19. }

代码示例来源:origin: mccorby/FederatedAndroidTrainer

  1. @Override
  2. public void onNext(@NonNull byte[] gradient) {
  3. try {
  4. INDArray remoteGradient = Nd4j.fromByteArray(gradient);
  5. for (FederatedModel model: models) {
  6. model.updateWeights(remoteGradient);
  7. }
  8. } catch (IOException e) {
  9. e.printStackTrace();
  10. }
  11. view.onGradientReceived(gradient);
  12. }

相关文章