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

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

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

Nd4j.getNDArrayFactory介绍

暂无

代码示例

代码示例来源:origin: deeplearning4j/nd4j

  1. /**
  2. * Sort an ndarray along a particular dimension<br>
  3. * Note that the input array is modified in-place.
  4. *
  5. * @param ndarray the ndarray to sort
  6. * @param dimension the dimension to sort
  7. * @return the sorted ndarray
  8. */
  9. public static INDArray sort(INDArray ndarray, int dimension, boolean ascending) {
  10. return getNDArrayFactory().sort(ndarray, !ascending, dimension);
  11. }

代码示例来源:origin: deeplearning4j/nd4j

  1. public static INDArray sort(INDArray ndarray, boolean ascending) {
  2. return getNDArrayFactory().sort(ndarray, !ascending);
  3. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer compress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(getBufferTypeEx(buffer), buffer,
  4. DataBuffer.TypeEx.FLOAT16);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer compress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(getBufferTypeEx(buffer), buffer,
  4. DataBuffer.TypeEx.INT16);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer decompress(DataBuffer buffer) {
  3. val type = getGlobalTypeEx();
  4. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(DataBuffer.TypeEx.FLOAT16, buffer, type);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer decompress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(DataBuffer.TypeEx.UINT8, buffer, getGlobalTypeEx());
  4. return result;
  5. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer compress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(getBufferTypeEx(buffer), buffer,
  4. DataBuffer.TypeEx.UINT8);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer compress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(getBufferTypeEx(buffer), buffer,
  4. DataBuffer.TypeEx.FLOAT8);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer decompress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(DataBuffer.TypeEx.INT8, buffer, getGlobalTypeEx());
  4. return result;
  5. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer decompress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(DataBuffer.TypeEx.FLOAT8, buffer, getGlobalTypeEx());
  4. return result;
  5. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer compress(DataBuffer buffer) {
  3. DataBuffer result =
  4. Nd4j.getNDArrayFactory().convertDataEx(getBufferTypeEx(buffer), buffer, DataBuffer.TypeEx.INT8);
  5. return result;
  6. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public DataBuffer decompress(DataBuffer buffer) {
  3. DataBuffer result = Nd4j.getNDArrayFactory().convertDataEx(DataBuffer.TypeEx.INT16, buffer, getGlobalTypeEx());
  4. return result;
  5. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public INDArray percentile(Number quantile, int... dimension) {
  3. if (quantile.doubleValue() < 0 || quantile.doubleValue() > 100)
  4. throw new ND4JIllegalStateException("Percentile value should be in 0...100 range");
  5. if (isScalar())
  6. return Nd4j.scalar(this.getDouble(0));
  7. INDArray sorted = Nd4j.getNDArrayFactory().sort(this.dup(this.ordering()), false, dimension);
  8. // there's no practical sense doing this on GPU, stride will be just size of TAD.
  9. INDArray ret = Nd4j.createUninitialized(sorted.tensorssAlongDimension(dimension));
  10. for (int i = 0; i < ret.length(); i++) {
  11. ret.putScalar(i, getPercentile(quantile, sorted.tensorAlongDimension(i, dimension)));
  12. }
  13. return ret;
  14. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public INDArray percentile(Number quantile, int... dimension) {
  3. if (quantile.doubleValue() < 0 || quantile.doubleValue() > 100)
  4. throw new ND4JIllegalStateException("Percentile value should be in 0...100 range");
  5. if (isScalar())
  6. return Nd4j.scalar(this.getDouble(0));
  7. INDArray sorted = Nd4j.getNDArrayFactory().sort(this.dup(this.ordering()), false, dimension);
  8. // there's no practical sense doing this on GPU, stride will be just size of TAD.
  9. INDArray ret = Nd4j.createUninitialized(sorted.tensorssAlongDimension(dimension));
  10. for (int i = 0; i < ret.length(); i++) {
  11. ret.putScalar(i, getPercentile(quantile, sorted.tensorAlongDimension(i, dimension)));
  12. }
  13. return ret;
  14. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public INDArray convertToDoubles() {
  3. if (data.dataType() == DataBuffer.Type.DOUBLE)
  4. return this;
  5. val factory = Nd4j.getNDArrayFactory();
  6. val buffer = Nd4j.createBuffer(new long[]{this.length()}, DataBuffer.Type.DOUBLE);
  7. factory.convertDataEx(convertType(data.dataType()), this.data().addressPointer(), DataBuffer.TypeEx.DOUBLE, buffer.addressPointer(), buffer.length());
  8. return Nd4j.createArrayFromShapeBuffer(buffer, this.shapeInformation);
  9. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. public INDArray convertToFloats() {
  3. if (data.dataType() == DataBuffer.Type.FLOAT)
  4. return this;
  5. val factory = Nd4j.getNDArrayFactory();
  6. val buffer = Nd4j.createBuffer(new long[]{this.length()}, DataBuffer.Type.FLOAT);
  7. factory.convertDataEx(convertType(data.dataType()), this.data().addressPointer(), DataBuffer.TypeEx.FLOAT, buffer.addressPointer(), buffer.length());
  8. return Nd4j.createArrayFromShapeBuffer(buffer, this.shapeInformation);
  9. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. protected CompressedDataBuffer compressPointer(DataBuffer.TypeEx srcType, Pointer srcPointer, int length,
  3. int elementSize) {
  4. BytePointer ptr = new BytePointer(length);
  5. CompressionDescriptor descriptor = new CompressionDescriptor();
  6. descriptor.setCompressedLength(length * 1);
  7. descriptor.setOriginalLength(length * elementSize);
  8. descriptor.setOriginalElementSize(elementSize);
  9. descriptor.setNumberOfElements(length);
  10. descriptor.setCompressionAlgorithm(getDescriptor());
  11. descriptor.setCompressionType(getCompressionType());
  12. CompressedDataBuffer buffer = new CompressedDataBuffer(ptr, descriptor);
  13. Nd4j.getNDArrayFactory().convertDataEx(srcType, srcPointer, DataBuffer.TypeEx.UINT8, ptr, length);
  14. return buffer;
  15. }
  16. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. protected CompressedDataBuffer compressPointer(DataBuffer.TypeEx srcType, Pointer srcPointer, int length,
  3. int elementSize) {
  4. BytePointer ptr = new BytePointer(length);
  5. CompressionDescriptor descriptor = new CompressionDescriptor();
  6. descriptor.setCompressedLength(length * 1);
  7. descriptor.setOriginalLength(length * elementSize);
  8. descriptor.setOriginalElementSize(elementSize);
  9. descriptor.setNumberOfElements(length);
  10. descriptor.setCompressionAlgorithm(getDescriptor());
  11. descriptor.setCompressionType(getCompressionType());
  12. CompressedDataBuffer buffer = new CompressedDataBuffer(ptr, descriptor);
  13. Nd4j.getNDArrayFactory().convertDataEx(srcType, srcPointer, DataBuffer.TypeEx.FLOAT8, ptr, length);
  14. return buffer;
  15. }
  16. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. protected CompressedDataBuffer compressPointer(DataBuffer.TypeEx srcType, Pointer srcPointer, int length,
  3. int elementSize) {
  4. BytePointer ptr = new BytePointer(length * 2);
  5. CompressionDescriptor descriptor = new CompressionDescriptor();
  6. descriptor.setCompressedLength(length * 2);
  7. descriptor.setOriginalLength(length * elementSize);
  8. descriptor.setOriginalElementSize(elementSize);
  9. descriptor.setNumberOfElements(length);
  10. descriptor.setCompressionAlgorithm(getDescriptor());
  11. descriptor.setCompressionType(getCompressionType());
  12. CompressedDataBuffer buffer = new CompressedDataBuffer(ptr, descriptor);
  13. Nd4j.getNDArrayFactory().convertDataEx(srcType, srcPointer, DataBuffer.TypeEx.INT16, ptr, length);
  14. return buffer;
  15. }
  16. }

代码示例来源:origin: deeplearning4j/nd4j

  1. @Override
  2. protected CompressedDataBuffer compressPointer(DataBuffer.TypeEx srcType, Pointer srcPointer, int length,
  3. int elementSize) {
  4. BytePointer ptr = new BytePointer(length * 2);
  5. CompressionDescriptor descriptor = new CompressionDescriptor();
  6. descriptor.setCompressedLength(length * 2);
  7. descriptor.setOriginalLength(length * elementSize);
  8. descriptor.setOriginalElementSize(elementSize);
  9. descriptor.setNumberOfElements(length);
  10. descriptor.setCompressionAlgorithm(getDescriptor());
  11. descriptor.setCompressionType(getCompressionType());
  12. CompressedDataBuffer buffer = new CompressedDataBuffer(ptr, descriptor);
  13. Nd4j.getNDArrayFactory().convertDataEx(srcType, srcPointer, DataBuffer.TypeEx.FLOAT16, ptr, length);
  14. return buffer;
  15. }
  16. }

相关文章