org.nd4j.linalg.jblas

Class BlasWrapper

java.lang.Object

org.nd4j.linalg.jblas.BlasWrapper

All Implemented Interfaces:

BlasWrapper

public class BlasWrapper
extends Object
implements BlasWrapper

Copy of SimpleBlas to handle offsets implementing an interface for library neutral jblas operations

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
BlasWrapper()

Method Summary

Modifier and Type	Method and Description
double	asum(IComplexNDArray x)
double	asum(INDArray x) Compute || x ||_1 (1-norm, sum of absolute values)
INDArray	axpy(double da, INDArray dx, INDArray dy) Compute y <- alpha * x + y (elementwise addition)
INDArray	axpy(float da, INDArray dx, INDArray dy) Compute y <- alpha * x + y (elementwise addition)
IComplexNDArray	axpy(IComplexNumber da, IComplexNDArray dx, IComplexNDArray dy)
void	checkInfo(String name, int info)
IComplexNDArray	copy(IComplexNDArray x, IComplexNDArray y)
INDArray	copy(INDArray x, INDArray y) Compute y <- x (copy a matrix)
void	dcopy(int n, float[] dx, int dxIdx, int incx, float[] dy, int dyIdx, int incy)
double	dot(INDArray x, INDArray y) Compute x^T * y (dot product)
IComplexNumber	dotc(IComplexNDArray x, IComplexNDArray y) Compute x^T * y (dot product)
IComplexNumber	dotu(IComplexNDArray x, IComplexNDArray y) Compute x^T * y (dot product)
int	geev(char jobvl, char jobvr, INDArray A, INDArray WR, INDArray WI, INDArray VL, INDArray VR)
void	gelsd(INDArray A, INDArray B) Generalized Least Squares via *GELSD.
INDArray	gemm(double alpha, INDArray a, INDArray b, double beta, INDArray c) Compute c <- a*b + beta * c (general matrix matrix multiplication)
INDArray	gemm(float alpha, INDArray a, INDArray b, float beta, INDArray c) Compute c <- a*b + beta * c (general matrix matrix multiplication)
IComplexNDArray	gemm(IComplexNumber alpha, IComplexNDArray a, IComplexNDArray b, IComplexNumber beta, IComplexNDArray c)
INDArray	gemv(double alpha, INDArray a, INDArray x, double beta, INDArray y) Compute y <- alpha*op(a)*x + beta * y (general matrix vector multiplication)
INDArray	gemv(float alpha, INDArray a, INDArray x, float beta, INDArray y) Compute y <- alpha*op(a)*x + beta * y (general matrix vector multiplication)
IComplexNDArray	gemv(IComplexDouble alpha, IComplexNDArray a, IComplexNDArray x, IComplexDouble beta, IComplexNDArray y)
IComplexNDArray	gemv(IComplexFloat alpha, IComplexNDArray a, IComplexNDArray x, IComplexFloat beta, IComplexNDArray y)
void	geqrf(INDArray A, INDArray tau)
INDArray	ger(double alpha, INDArray x, INDArray y, INDArray a) Compute A <- alpha * x * y^T + A (general rank-1 update)
INDArray	ger(float alpha, INDArray x, INDArray y, INDArray a) Compute A <- alpha * x * y^T + A (general rank-1 update)
IComplexNDArray	gerc(IComplexDouble alpha, IComplexNDArray x, IComplexNDArray y, IComplexNDArray a) Compute A <- alpha * x * y^H + A (general rank-1 update)
IComplexNDArray	gerc(IComplexFloat alpha, IComplexNDArray x, IComplexNDArray y, IComplexNDArray a) Compute A <- alpha * x * y^H + A (general rank-1 update)
IComplexNDArray	geru(IComplexDouble alpha, IComplexNDArray x, IComplexNDArray y, IComplexNDArray a) Compute A <- alpha * x * y^T + A (general rank-1 update)
IComplexNDArray	geru(IComplexFloat alpha, IComplexNDArray x, IComplexNDArray y, IComplexNDArray a) Compute A <- alpha * x * y^T + A (general rank-1 update)
INDArray	gesv(INDArray a, int[] ipiv, INDArray b) LAPACK
int	iamax(IComplexNDArray x) Compute index of element with largest absolute value (complex version).
int	iamax(INDArray x) Compute index of element with largest absolute value (index of absolute value maximum)
double	nrm2(IComplexNDArray x)
double	nrm2(INDArray x) Compute || x ||_2 (2-norm)
void	ormqr(char side, char trans, INDArray A, INDArray tau, INDArray C)
void	posv(char uplo, INDArray A, INDArray B)
void	saxpy(double alpha, INDArray x, INDArray y) Abstraction over saxpy
void	saxpy(float alpha, INDArray x, INDArray y) Abstraction over saxpy
INDArray	scal(double alpha, INDArray x) Compute x <- alpha * x (scale a matrix)
INDArray	scal(float alpha, INDArray x) Compute x <- alpha * x (scale a matrix)
IComplexNDArray	scal(IComplexDouble alpha, IComplexNDArray x)
IComplexNDArray	scal(IComplexFloat alpha, IComplexNDArray x)
INDArray	swap(INDArray x, INDArray y) Compute x <-> y (swap two matrices)
int	syev(char jobz, char uplo, INDArray a, INDArray w)
int	syevd(char jobz, char uplo, INDArray A, INDArray w)
int	syevr(char jobz, char range, char uplo, INDArray a, double vl, double vu, int il, int iu, double abstol, INDArray w, INDArray z, int[] isuppz)
int	syevr(char jobz, char range, char uplo, INDArray a, float vl, float vu, int il, int iu, float abstol, INDArray w, INDArray z, int[] isuppz)
int	syevx(char jobz, char range, char uplo, INDArray a, double vl, double vu, int il, int iu, double abstol, INDArray w, INDArray z)
int	syevx(char jobz, char range, char uplo, INDArray a, float vl, float vu, int il, int iu, float abstol, INDArray w, INDArray z)
int	sygvd(int itype, char jobz, char uplo, INDArray A, INDArray B, INDArray W)
INDArray	sysv(char uplo, INDArray a, int[] ipiv, INDArray b)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

BlasWrapper

public BlasWrapper()

Method Detail

swap

public INDArray swap(INDArray x,
                     INDArray y)

Compute x <-> y (swap two matrices)

Specified by:

swap in interface BlasWrapper

scal

public INDArray scal(double alpha,
                     INDArray x)

Compute x <- alpha * x (scale a matrix)

Specified by:

scal in interface BlasWrapper

scal

public INDArray scal(float alpha,
                     INDArray x)

Compute x <- alpha * x (scale a matrix)

Specified by:

scal in interface BlasWrapper

scal

public IComplexNDArray scal(IComplexFloat alpha,
                            IComplexNDArray x)

Specified by:

scal in interface BlasWrapper

scal

public IComplexNDArray scal(IComplexDouble alpha,
                            IComplexNDArray x)

Specified by:

scal in interface BlasWrapper

copy

public INDArray copy(INDArray x,
                     INDArray y)

Compute y <- x (copy a matrix)

Specified by:

copy in interface BlasWrapper

copy

public IComplexNDArray copy(IComplexNDArray x,
                            IComplexNDArray y)

Specified by:

copy in interface BlasWrapper

axpy

public INDArray axpy(double da,
                     INDArray dx,
                     INDArray dy)

Compute y <- alpha * x + y (elementwise addition)

Specified by:

axpy in interface BlasWrapper

axpy

public INDArray axpy(float da,
                     INDArray dx,
                     INDArray dy)

Compute y <- alpha * x + y (elementwise addition)

Specified by:

axpy in interface BlasWrapper

axpy

public IComplexNDArray axpy(IComplexNumber da,
                            IComplexNDArray dx,
                            IComplexNDArray dy)

Specified by:

axpy in interface BlasWrapper

dot

public double dot(INDArray x,
                  INDArray y)

Compute x^T * y (dot product)

Specified by:

dot in interface BlasWrapper

dotc

public IComplexNumber dotc(IComplexNDArray x,
                           IComplexNDArray y)

Compute x^T * y (dot product)

Specified by:

dotc in interface BlasWrapper

dotu

public IComplexNumber dotu(IComplexNDArray x,
                           IComplexNDArray y)

Compute x^T * y (dot product)

Specified by:

dotu in interface BlasWrapper

nrm2

public double nrm2(INDArray x)

Compute || x ||_2 (2-norm)

Specified by:

nrm2 in interface BlasWrapper

nrm2

public double nrm2(IComplexNDArray x)

Specified by:

nrm2 in interface BlasWrapper

asum

public double asum(INDArray x)

Compute || x ||_1 (1-norm, sum of absolute values)

Specified by:

asum in interface BlasWrapper

asum

public double asum(IComplexNDArray x)

Specified by:

asum in interface BlasWrapper

iamax

public int iamax(INDArray x)

Compute index of element with largest absolute value (index of absolute value maximum)

Specified by:

iamax in interface BlasWrapper

iamax

public int iamax(IComplexNDArray x)

Compute index of element with largest absolute value (complex version).

Specified by:

iamax in interface BlasWrapper

Parameters:

x - matrix

Returns:

index of element with largest absolute value.

gemv

public INDArray gemv(double alpha,
                     INDArray a,
                     INDArray x,
                     double beta,
                     INDArray y)

Compute y <- alpha*op(a)*x + beta * y (general matrix vector multiplication)

Specified by:

gemv in interface BlasWrapper

gemv

public INDArray gemv(float alpha,
                     INDArray a,
                     INDArray x,
                     float beta,
                     INDArray y)

Compute y <- alpha*op(a)*x + beta * y (general matrix vector multiplication)

Specified by:

gemv in interface BlasWrapper

ger

public INDArray ger(double alpha,
                    INDArray x,
                    INDArray y,
                    INDArray a)

Compute A <- alpha * x * y^T + A (general rank-1 update)

Specified by:

ger in interface BlasWrapper

ger

public INDArray ger(float alpha,
                    INDArray x,
                    INDArray y,
                    INDArray a)

Compute A <- alpha * x * y^T + A (general rank-1 update)

Specified by:

ger in interface BlasWrapper

gemv

public IComplexNDArray gemv(IComplexDouble alpha,
                            IComplexNDArray a,
                            IComplexNDArray x,
                            IComplexDouble beta,
                            IComplexNDArray y)

Specified by:

gemv in interface BlasWrapper

gemv

public IComplexNDArray gemv(IComplexFloat alpha,
                            IComplexNDArray a,
                            IComplexNDArray x,
                            IComplexFloat beta,
                            IComplexNDArray y)

Specified by:

gemv in interface BlasWrapper

geru

public IComplexNDArray geru(IComplexDouble alpha,
                            IComplexNDArray x,
                            IComplexNDArray y,
                            IComplexNDArray a)

Compute A <- alpha * x * y^T + A (general rank-1 update)

Specified by:

geru in interface BlasWrapper

Parameters:

alpha -

x -

y -

a -

geru

public IComplexNDArray geru(IComplexFloat alpha,
                            IComplexNDArray x,
                            IComplexNDArray y,
                            IComplexNDArray a)

Compute A <- alpha * x * y^T + A (general rank-1 update)

Specified by:

geru in interface BlasWrapper

Parameters:

alpha -

x -

y -

a -

gerc

public IComplexNDArray gerc(IComplexDouble alpha,
                            IComplexNDArray x,
                            IComplexNDArray y,
                            IComplexNDArray a)

Compute A <- alpha * x * y^H + A (general rank-1 update)

Specified by:

gerc in interface BlasWrapper

gerc

public IComplexNDArray gerc(IComplexFloat alpha,
                            IComplexNDArray x,
                            IComplexNDArray y,
                            IComplexNDArray a)

Compute A <- alpha * x * y^H + A (general rank-1 update)

Specified by:

gerc in interface BlasWrapper

gemm

public INDArray gemm(double alpha,
                     INDArray a,
                     INDArray b,
                     double beta,
                     INDArray c)

Compute c <- a*b + beta * c (general matrix matrix multiplication)

Specified by:

gemm in interface BlasWrapper

gemm

public INDArray gemm(float alpha,
                     INDArray a,
                     INDArray b,
                     float beta,
                     INDArray c)

Compute c <- a*b + beta * c (general matrix matrix multiplication)

Specified by:

gemm in interface BlasWrapper

gemm

public IComplexNDArray gemm(IComplexNumber alpha,
                            IComplexNDArray a,
                            IComplexNDArray b,
                            IComplexNumber beta,
                            IComplexNDArray c)

Specified by:

gemm in interface BlasWrapper

gesv

public INDArray gesv(INDArray a,
                     int[] ipiv,
                     INDArray b)

LAPACK

Specified by:

gesv in interface BlasWrapper

checkInfo

public void checkInfo(String name,
                      int info)

Specified by:

checkInfo in interface BlasWrapper

sysv

public INDArray sysv(char uplo,
                     INDArray a,
                     int[] ipiv,
                     INDArray b)

Specified by:

sysv in interface BlasWrapper

syev

public int syev(char jobz,
                char uplo,
                INDArray a,
                INDArray w)

Specified by:

syev in interface BlasWrapper

syevx

public int syevx(char jobz,
                 char range,
                 char uplo,
                 INDArray a,
                 double vl,
                 double vu,
                 int il,
                 int iu,
                 double abstol,
                 INDArray w,
                 INDArray z)

Specified by:

syevx in interface BlasWrapper

syevx

public int syevx(char jobz,
                 char range,
                 char uplo,
                 INDArray a,
                 float vl,
                 float vu,
                 int il,
                 int iu,
                 float abstol,
                 INDArray w,
                 INDArray z)

Specified by:

syevx in interface BlasWrapper

syevd

public int syevd(char jobz,
                 char uplo,
                 INDArray A,
                 INDArray w)

Specified by:

syevd in interface BlasWrapper

syevr

public int syevr(char jobz,
                 char range,
                 char uplo,
                 INDArray a,
                 double vl,
                 double vu,
                 int il,
                 int iu,
                 double abstol,
                 INDArray w,
                 INDArray z,
                 int[] isuppz)

Specified by:

syevr in interface BlasWrapper

syevr

public int syevr(char jobz,
                 char range,
                 char uplo,
                 INDArray a,
                 float vl,
                 float vu,
                 int il,
                 int iu,
                 float abstol,
                 INDArray w,
                 INDArray z,
                 int[] isuppz)

Specified by:

syevr in interface BlasWrapper

posv

public void posv(char uplo,
                 INDArray A,
                 INDArray B)

Specified by:

posv in interface BlasWrapper

geev

public int geev(char jobvl,
                char jobvr,
                INDArray A,
                INDArray WR,
                INDArray WI,
                INDArray VL,
                INDArray VR)

Specified by:

geev in interface BlasWrapper

sygvd

public int sygvd(int itype,
                 char jobz,
                 char uplo,
                 INDArray A,
                 INDArray B,
                 INDArray W)

Specified by:

sygvd in interface BlasWrapper

gelsd

public void gelsd(INDArray A,
                  INDArray B)

Generalized Least Squares via *GELSD. Note that B must be padded to contain the solution matrix. This occurs when A has fewer rows than columns. For example: in A * X = B, A is (m,n), X is (n,k) and B is (m,k). Now if m < n, since B is overwritten to contain the solution (in classical LAPACK style), B needs to be padded to be an (n,k) matrix. Likewise, if m > n, the solution consists only of the first n rows of B.

Specified by:

gelsd in interface BlasWrapper

Parameters:

A - an (m,n) matrix

B - an (max(m,n), k) matrix (well, at least)

geqrf

public void geqrf(INDArray A,
                  INDArray tau)

Specified by:

geqrf in interface BlasWrapper

ormqr

public void ormqr(char side,
                  char trans,
                  INDArray A,
                  INDArray tau,
                  INDArray C)

Specified by:

ormqr in interface BlasWrapper

dcopy

public void dcopy(int n,
                  float[] dx,
                  int dxIdx,
                  int incx,
                  float[] dy,
                  int dyIdx,
                  int incy)

Specified by:

dcopy in interface BlasWrapper

saxpy

public void saxpy(double alpha,
                  INDArray x,
                  INDArray y)

Abstraction over saxpy

Specified by:

saxpy in interface BlasWrapper

Parameters:

alpha - the alpha to scale by

x - the ndarray to use

y - the ndarray to use

saxpy

public void saxpy(float alpha,
                  INDArray x,
                  INDArray y)

Abstraction over saxpy

Specified by:

saxpy in interface BlasWrapper

Parameters:

alpha - the alpha to scale by

x - the ndarray to use

y - the ndarray to use