@Override public DataBuffer createFloat(long offset, int[] data) { FloatBuffer ret = new FloatBuffer(ArrayUtil.toFloats(data), true, offset); return ret; }
@Override public DataBuffer createFloat(long offset, int[] data, boolean copy) { return new FloatBuffer(ArrayUtil.toFloats(data), copy, offset); }
@Override public DataBuffer createFloat(double[] data, boolean copy) { return new FloatBuffer(ArrayUtil.toFloats(data), copy); }
@Override public DataBuffer createFloat(long offset, double[] data) { return new FloatBuffer(ArrayUtil.toFloats(data), true, offset); }
@Override public DataBuffer createFloat(int[] data, boolean copy) { return new FloatBuffer(ArrayUtil.toFloats(data), copy); }
@Override public DataBuffer createFloat(long offset, double[] data, boolean copy) { return new FloatBuffer(ArrayUtil.toFloats(data), copy, offset); }
public static float[] toFloats(int[][] ints) { return toFloats(Ints.concat(ints)); }
public static DataBuffer createBufferDetached(double[] data) { DataBuffer ret; if (dataType() == DataBuffer.Type.DOUBLE) ret = DATA_BUFFER_FACTORY_INSTANCE.createDouble(data); else if (dataType() == DataBuffer.Type.HALF) ret = DATA_BUFFER_FACTORY_INSTANCE.createHalf(ArrayUtil.toFloats(data)); else ret = DATA_BUFFER_FACTORY_INSTANCE.createFloat(ArrayUtil.toFloats(data)); logCreationIfNecessary(ret); return ret; }
public static INDArray toNDArray(int[] nums) { if (Nd4j.dataType() == DataBuffer.Type.DOUBLE) { double[] doubles = ArrayUtil.toDoubles(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } else { float[] doubles = ArrayUtil.toFloats(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } }
public static INDArray toNDArray(long[] nums) { if (Nd4j.dataType() == DataBuffer.Type.DOUBLE) { double[] doubles = ArrayUtil.toDoubles(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } else { float[] doubles = ArrayUtil.toFloats(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } }
public static INDArray toNDArray(int[][] nums) { if (Nd4j.dataType() == DataBuffer.Type.DOUBLE) { double[] doubles = ArrayUtil.toDoubles(nums); INDArray create = Nd4j.create(doubles, new int[] {nums[0].length, nums.length}); return create; } else { float[] doubles = ArrayUtil.toFloats(nums); INDArray create = Nd4j.create(doubles, new int[] {nums[0].length, nums.length}); return create; } }
/** * Create a buffer based on the data opType * * @param data the data to create the buffer with * @return the created buffer */ public static DataBuffer createBuffer(double[] data, long offset) { DataBuffer ret; if (dataType() == DataBuffer.Type.DOUBLE) ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createDouble(offset, data) : DATA_BUFFER_FACTORY_INSTANCE.createDouble(offset, data, Nd4j.getMemoryManager().getCurrentWorkspace()); else if (dataType() == DataBuffer.Type.HALF) ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createHalf(offset, data) : DATA_BUFFER_FACTORY_INSTANCE.createHalf(offset, ArrayUtil.toFloats(data), Nd4j.getMemoryManager().getCurrentWorkspace()); else ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createFloat(offset, ArrayUtil.toFloats(data)) : DATA_BUFFER_FACTORY_INSTANCE.createFloat(offset, ArrayUtil.toFloats(data), Nd4j.getMemoryManager().getCurrentWorkspace()); logCreationIfNecessary(ret); return ret; }
/** * Create a buffer based on the data opType * * @param data the data to create the buffer with * @return the created buffer */ public static DataBuffer createBuffer(double[] data) { DataBuffer ret; if (dataType() == DataBuffer.Type.DOUBLE) ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createDouble(data) : DATA_BUFFER_FACTORY_INSTANCE.createDouble(data, Nd4j.getMemoryManager().getCurrentWorkspace()); else if (dataType() == DataBuffer.Type.HALF) ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createHalf(data) : DATA_BUFFER_FACTORY_INSTANCE.createHalf(ArrayUtil.toFloats(data), Nd4j.getMemoryManager().getCurrentWorkspace()); else ret = Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createFloat(ArrayUtil.toFloats(data)) : DATA_BUFFER_FACTORY_INSTANCE.createFloat(ArrayUtil.toFloats(data), Nd4j.getMemoryManager().getCurrentWorkspace()); logCreationIfNecessary(ret); return ret; }
@Override public void resolvePropertiesFromSameDiffBeforeExecution() { super.resolvePropertiesFromSameDiffBeforeExecution(); if (broadcast != null && numInputArguments() < 2) { if (numInputArguments() == 0) { addInputArgument(args()[0].getArr(), Nd4j.create(ArrayUtil.toFloats(broadcast)).reshape(broadcast.length)); } else if (numInputArguments() == 1) { addInputArgument(Nd4j.create(ArrayUtil.toFloats(broadcast))); } } if (numIArguments() < 1) { addIArgument(axis); } if (numOutputArguments() < getDescriptor().getNumOutputs()) { val outputs = outputVariables(); for (int i = 0; i < outputs.length; i++) { val output = outputs[i].getArr(); addOutputArgument(output); } } }
@Override public DataBuffer createFloat(int[] data, boolean copy) { return new CudaFloatDataBuffer(ArrayUtil.toFloats(data)); }
@Override public DataBuffer createFloat(double[] data, boolean copy) { return new CudaFloatDataBuffer(ArrayUtil.toFloats(data)); }
@Override public DataBuffer createFloat(double[] data) { return new CudaFloatDataBuffer(ArrayUtil.toFloats(data)); }
@Override public DataBuffer createFloat(double[] data, boolean copy) { return new CudaFloatDataBuffer(ArrayUtil.toFloats(data)); }
public static DataBuffer createBufferDetached(double[] data) { DataBuffer ret; if (dataType() == DataBuffer.Type.DOUBLE) ret = DATA_BUFFER_FACTORY_INSTANCE.createDouble(data); else if (dataType() == DataBuffer.Type.HALF) ret = DATA_BUFFER_FACTORY_INSTANCE.createHalf(ArrayUtil.toFloats(data)); else ret = DATA_BUFFER_FACTORY_INSTANCE.createFloat(ArrayUtil.toFloats(data)); logCreationIfNecessary(ret); return ret; }
public static INDArray toNDArray(int[] nums) { if (Nd4j.dataType() == DataBuffer.Type.DOUBLE) { double[] doubles = ArrayUtil.toDoubles(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } else { float[] doubles = ArrayUtil.toFloats(nums); INDArray create = Nd4j.create(doubles, new int[] {1, nums.length}); return create; } }