org.nd4j.linalg.jcublas.JCublasNDArrayFactory Maven / Gradle / Ivy
/*
*
* * Copyright 2015 Skymind,Inc.
* *
* * Licensed under the Apache License, Version 2.0 (the "License");
* * you may not use this file except in compliance with the License.
* * You may obtain a copy of the License at
* *
* * http://www.apache.org/licenses/LICENSE-2.0
* *
* * Unless required by applicable law or agreed to in writing, software
* * distributed under the License is distributed on an "AS IS" BASIS,
* * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* * See the License for the specific language governing permissions and
* * limitations under the License.
*
*
*/
package org.nd4j.linalg.jcublas;
import org.nd4j.linalg.api.buffer.DataBuffer;
import org.nd4j.linalg.api.complex.IComplexDouble;
import org.nd4j.linalg.api.complex.IComplexFloat;
import org.nd4j.linalg.api.complex.IComplexNDArray;
import org.nd4j.linalg.api.complex.IComplexNumber;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.BaseNDArrayFactory;
import org.nd4j.linalg.factory.Nd4j;
import org.nd4j.linalg.jcublas.complex.ComplexDouble;
import org.nd4j.linalg.jcublas.complex.ComplexFloat;
import org.nd4j.linalg.jcublas.complex.JCublasComplexNDArray;
import org.nd4j.linalg.util.ArrayUtil;
import java.util.List;
/**
* Jcublas ndarray factory. Handles creation of
* jcuda.jcublas ndarrays.
*
* @author mjk
*/
public class JCublasNDArrayFactory extends BaseNDArrayFactory {
public JCublasNDArrayFactory() {
}
public JCublasNDArrayFactory(DataBuffer.Type dtype, Character order) {
super(dtype, order);
}
public JCublasNDArrayFactory(DataBuffer.Type dtype, char order) {
super(dtype, order);
}
@Override
public INDArray create(int[] shape, DataBuffer buffer) {
return new JCublasNDArray(shape, buffer);
}
/**
* Create float
*
* @param real real component
* @param imag imag component
* @return
*/
@Override
public IComplexFloat createFloat(float real, float imag) {
return new ComplexFloat(real, imag);
}
/**
* Create an instance of a complex double
*
* @param real the real component
* @param imag the imaginary component
* @return a new imaginary double with the specified real and imaginary components
*/
@Override
public IComplexDouble createDouble(double real, double imag) {
return new ComplexDouble(real, imag);
}
/**
* Create an ndarray with the given data layout
*
* @param data the data to create the ndarray with
* @return the ndarray with the given data layout
*/
@Override
public INDArray create(double[][] data) {
return new JCublasNDArray(data);
}
/**
* Create a complex ndarray from the passed in indarray
*
* @param arr the arr to wrap
* @return the complex ndarray with the specified ndarray as the
* real components
*/
@Override
public IComplexNDArray createComplex(INDArray arr) {
return new JCublasComplexNDArray(arr);
}
/**
* Create a complex ndarray from the passed in indarray
*
* @param data the data to wrap
* @param shape
* @return the complex ndarray with the specified ndarray as the
* real components
*/
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape) {
return new JCublasComplexNDArray(data, shape,Nd4j.getComplexStrides(shape,Nd4j.order()));
}
/**
* Create a complex ndarray from the passed in indarray
*
* @param arrs the arr to wrap
* @param shape
* @return the complex ndarray with the specified ndarray as the
* real components
*/
@Override
public IComplexNDArray createComplex(List arrs, int[] shape) {
return new JCublasComplexNDArray(arrs, shape);
}
@Override
public INDArray create(DataBuffer data) {
return new JCublasNDArray(data);
}
@Override
public IComplexNDArray createComplex(DataBuffer data) {
return new JCublasComplexNDArray(data);
}
@Override
public IComplexNDArray createComplex(DataBuffer data, int rows, int columns, int[] stride, int offset) {
return new JCublasComplexNDArray(data, new int[]{rows, columns}, stride, offset);
}
@Override
public INDArray create(DataBuffer data, int rows, int columns, int[] stride, int offset) {
return new JCublasNDArray(data, new int[]{rows, columns}, stride, offset);
}
@Override
public IComplexNDArray createComplex(DataBuffer data, int[] shape, int[] stride, int offset) {
return new JCublasComplexNDArray(data, shape, stride, offset);
}
/**
* Creates a complex ndarray with the specified shape
*
* @param data the data to use with the ndarray
* @param shape the shape of the ndarray
* @param stride the stride for the ndarray
* @param offset the offset of the ndarray
* @return the instance
*/
@Override
public IComplexNDArray createComplex(float[] data, int[] shape, int[] stride, int offset) {
return new JCublasComplexNDArray(data, shape, stride, offset);
}
@Override
public INDArray create(int[] shape, char ordering) {
return new JCublasNDArray(shape, ordering);
}
@Override
public INDArray create(DataBuffer data, int[] newShape, int[] newStride, int offset, char ordering) {
return new JCublasNDArray(data, newShape, newStride, offset, ordering);
}
@Override
public IComplexNDArray createComplex(DataBuffer data, int[] newDims, int[] newStrides, int offset, char ordering) {
return new JCublasComplexNDArray(data, newDims, newStrides, offset, ordering);
}
@Override
public IComplexNDArray createComplex(float[] data, Character order) {
return new JCublasComplexNDArray(data, order);
}
@Override
public INDArray create(float[] data, int[] shape, int offset, Character order) {
return new JCublasNDArray(data, shape, offset, order);
}
@Override
public INDArray create(float[] data, int rows, int columns, int[] stride, int offset, char ordering) {
return new JCublasNDArray(data, new int[]{rows, columns}, stride, offset, ordering);
}
@Override
public INDArray create(double[] data, int[] shape, char ordering) {
return new JCublasNDArray(data, shape, ordering);
}
@Override
public INDArray create(List list, int[] shape, char ordering) {
return new JCublasNDArray(list, shape, ordering);
}
@Override
public INDArray create(double[] data, int[] shape, int offset) {
return new JCublasNDArray(data, shape, (char) offset);
}
@Override
public INDArray create(double[] data, int[] shape, int[] stride, int offset, char ordering) {
return new JCublasNDArray(data, shape, stride, offset, ordering);
}
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape, int[] stride, int offset) {
return new JCublasComplexNDArray(data, shape, stride, offset);
}
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape, int[] stride, int offset, char ordering) {
return new JCublasComplexNDArray(data, shape, stride, offset, ordering);
}
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape, int[] stride, char ordering) {
return new JCublasComplexNDArray(data, shape, stride, ordering);
}
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape, int offset, char ordering) {
return new JCublasComplexNDArray(data, shape, offset, ordering);
}
@Override
public IComplexNDArray createComplex(IComplexNumber[] data, int[] shape, char ordering) {
return new JCublasComplexNDArray(data, shape, ordering);
}
/**
* Creates an ndarray with the specified shape
*
* @param data
* @param shape the shape of the ndarray
* @param stride the stride for the ndarray
* @param offset the offset of the ndarray
* @return the instance
*/
@Override
public INDArray create(float[] data, int[] shape, int[] stride, int offset) {
return new JCublasNDArray(data, shape, stride, offset);
}
/**
* Creates a complex ndarray with the specified shape
*
* @param data
* @param shape the shape of the ndarray
* @param stride the stride for the ndarray
* @param offset the offset of the ndarray
* @return the instance
*/
@Override
public IComplexNDArray createComplex(double[] data, int[] shape, int[] stride, int offset) {
return new JCublasComplexNDArray(ArrayUtil.floatCopyOf(data), shape, stride, offset);
}
/**
* Creates an ndarray with the specified shape
*
* @param data
* @param shape the shape of the ndarray
* @param stride the stride for the ndarray
* @param offset the offset of the ndarray
* @return the instance
*/
@Override
public INDArray create(double[] data, int[] shape, int[] stride, int offset) {
return new JCublasNDArray(data, shape, stride, offset);
}
@Override
public INDArray create(DataBuffer data, int[] shape) {
return new JCublasNDArray(data, shape);
}
@Override
public IComplexNDArray createComplex(DataBuffer data, int[] shape) {
return new JCublasComplexNDArray(data, shape);
}
@Override
public IComplexNDArray createComplex(DataBuffer data, int[] shape, int[] stride) {
return new JCublasComplexNDArray(data, shape, stride);
}
@Override
public INDArray create(DataBuffer data, int[] shape, int[] stride, int offset) {
return new JCublasNDArray(data, shape, stride, offset);
}
/**
* Creates an ndarray with the specified shape
*
* @param list
* @param shape the shape of the ndarray
* @return the instance
*/
@Override
public INDArray create(List list, int[] shape) {
if (order == FORTRAN)
return new JCublasNDArray(list, shape, ArrayUtil.calcStridesFortran(shape));
else
return new JCublasNDArray(list, shape);
}
@Override
public IComplexNDArray createComplex(double[] data, int[] shape, int[] stride, int offset, char ordering) {
return new JCublasComplexNDArray(ArrayUtil.floatCopyOf(data), shape, stride, offset, ordering);
}
@Override
public IComplexNDArray createComplex(double[] data, int[] shape, int offset, char ordering) {
return new JCublasComplexNDArray(ArrayUtil.floatCopyOf(data), shape, offset, ordering);
}
@Override
public IComplexNDArray createComplex(DataBuffer buffer, int[] shape, int offset, char ordering) {
return new JCublasComplexNDArray(buffer, shape, offset, ordering);
}
@Override
public IComplexNDArray createComplex(double[] data, int[] shape, int offset) {
return new JCublasComplexNDArray(ArrayUtil.floatCopyOf(data), shape, offset);
}
@Override
public IComplexNDArray createComplex(DataBuffer buffer, int[] shape, int offset) {
return new JCublasComplexNDArray(buffer, shape, offset);
}
@Override
public INDArray create(float[] data, int[] shape, int offset) {
return new JCublasNDArray(data, shape, offset);
}
@Override
public IComplexNDArray createComplex(float[] data, int[] shape, int offset, char ordering) {
return new JCublasComplexNDArray(data, shape, Nd4j.getComplexStrides(shape, ordering), offset, ordering);
}
@Override
public IComplexNDArray createComplex(float[] data, int[] shape, int offset) {
return new JCublasComplexNDArray(data, shape, offset);
}
@Override
public IComplexNDArray createComplex(float[] data, int[] shape, int[] stride, int offset, char ordering) {
return new JCublasComplexNDArray(data, shape, stride, offset, ordering);
}
@Override
public INDArray create(float[][] floats) {
return new JCublasNDArray(floats);
}
@Override
public IComplexNDArray createComplex(float[] dim) {
if (dim.length % 2 != 0)
throw new IllegalArgumentException("Complex nd array buffers must have an even number of elements");
IComplexNDArray ret = Nd4j.createComplex(dim.length / 2);
int count = 0;
for (int i = 0; i < dim.length - 1; i += 2) {
ret.putScalar(count++, Nd4j.createDouble(dim[i], dim[i + 1]));
}
return ret;
}
@Override
public INDArray create(float[] data, int[] shape, int[] stride, int offset, char ordering) {
return new JCublasNDArray(data, shape, stride, offset, ordering);
}
@Override
public INDArray create(DataBuffer buffer, int[] shape, int offset) {
return new JCublasNDArray(buffer, shape, offset);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy