Java Code Examples for org.nd4j.linalg.factory.Nd4j#getComplexStrides()

@Override
public IComplexNDArray createComplex(float[] data, int[] shape, long offset, char ordering) {
    return new JCublasComplexNDArray(data, shape, Nd4j.getComplexStrides(shape, ordering), offset, ordering);
}
 

@Override
public IComplexNDArray createComplex(DataBuffer buffer, int[] shape, long offset, char ordering) {
    return new ComplexNDArray(buffer, shape, Nd4j.getComplexStrides(shape), offset, ordering);
}
 

@Override
public IComplexNDArray createComplex(DataBuffer buffer, int[] shape, long offset) {
    return new ComplexNDArray(buffer, shape, Nd4j.getComplexStrides(shape), offset, Nd4j.order());
}
 

public BaseComplexNDArray(long[] shape) {
    this(Nd4j.createBuffer(ArrayUtil.prodLong(shape) * 2), shape, Nd4j.getComplexStrides(shape));
}
 

public BaseComplexNDArray(int[] shape, char ordering) {
    this(Nd4j.createBuffer(ArrayUtil.prodLong(shape) * 2), shape, Nd4j.getComplexStrides(shape, ordering), 0,
                    ordering);
}
 

public BaseComplexNDArray(long[] shape, char ordering) {
    this(Nd4j.createBuffer(ArrayUtil.prodLong(shape) * 2), shape, Nd4j.getComplexStrides(shape, ordering), 0,
            ordering);
}
 

@Override
protected int[] getStrides(int[] shape, char ordering) {
    return Nd4j.getComplexStrides(shape, ordering);
}
 

/**
 * 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()));
}
 

/**
 * @param data
 * @param shape
 * @param offset
 * @param ordering
 * @return
 */
@Override
public IComplexNDArray createComplex(float[] data, int[] shape, long offset, char ordering) {
    return new ComplexNDArray(data, shape, Nd4j.getComplexStrides(shape, ordering), offset, ordering);

}
 

/**
 * Create this ndarray with the given data and shape and 0 offset
 *
 * @param data     the data to use
 * @param shape    the shape of the ndarray
 * @param ordering
 */
public BaseComplexNDArray(float[] data, int[] shape, char ordering) {
    this(data, shape, Nd4j.getComplexStrides(shape, ordering), 0, ordering);
}
 

/**
 *
 * @param shape
 * @param offset
 * @param ordering
 */
public BaseComplexNDArray(int[] shape, long offset, char ordering) {
    this(Nd4j.createBuffer(ArrayUtil.prodLong(shape) * 2), shape, Nd4j.getComplexStrides(shape, ordering), offset,
                    ordering);
}
 

/**
 *
 * @param shape
 */
public BaseComplexNDArray(int[] shape) {
    this(Nd4j.createBuffer(ArrayUtil.prodLong(shape) * 2), shape, Nd4j.getComplexStrides(shape));
}
 

/**
 * Create an ndarray from the specified slices
 * and the given shape
 *
 * @param slices   the slices of the ndarray
 * @param shape    the final shape of the ndarray
 * @param ordering the ordering of the ndarray
 */
public BaseComplexNDArray(List<IComplexNDArray> slices, int[] shape, char ordering) {
    this(slices, shape, Nd4j.getComplexStrides(shape, ordering), ordering);
}
 

/**
 *
 * @param data
 * @param shape
 * @param offset
 * @param ordering
 */
public BaseComplexNDArray(IComplexNumber[] data, int[] shape, long offset, char ordering) {
    this(data, shape, Nd4j.getComplexStrides(shape), offset, ordering);
}
 

/**
 *
 * @param buffer
 * @param shape
 * @param offset
 * @param ordering
 */
public BaseComplexNDArray(DataBuffer buffer, int[] shape, long offset, char ordering) {
    this(buffer, shape, Nd4j.getComplexStrides(shape), offset, ordering);
}
 

/**
 *
 * @param buffer
 * @param shape
 * @param offset
 */
public BaseComplexNDArray(DataBuffer buffer, int[] shape, long offset) {
    this(buffer, shape, Nd4j.getComplexStrides(shape), offset, Nd4j.order());
}