naga.array.Array Class Reference

Collaboration diagram for naga.array.Array:

Public Member Functions
def	__init__ (self, data=None, shape=None, dtype=None)
def	__repr__ (self)
def	__add__ (self, idata)
def	__sub__ (self, idata)
def	__mul__ (self, idata)
def	__getitem__ (self, indices)
def	copy (self)
def	dot (self, idata)
def	argmax (self)
def	max (self)
def	argmin (self)
def	min (self)
def	sum (self)
def	toarray (self)

Properties
	shape = property(lambda x: x.__shape)
	dtype = property(lambda x: x.__dtype)
	data = property(lambda x: x.__data)

Detailed Description

Definition at line 59 of file array.py.

Constructor & Destructor Documentation

__init__()

def naga.array.Array.__init__	(		self,
			data = None,
			shape = None,
			dtype = None
	)

Definition at line 61 of file array.py.

    def __init__(self, data=None, shape=None, dtype=None):
        if shape is not None:
            if dtype is None:
                dtype = np.float32
            if isinstance(shape, tuple):
                data = np.zeros(shape, dtype=dtype)
            else:
                raise TypeError("shape must be a tuple")
 
        if data is not None:
            if isinstance(data, list):
                data = np.array(data)
            if isinstance(data, np.ndarray):
                if dtype is not None:
                    data = data.astype(dtype)
                if data.dtype == np.int64 or data.dtype == np.int32:
                    self.__data = obj_int(context.context, data)
                elif data.dtype == np.float32:
                    self.__data = obj_float(context.context, data)
                elif data.dtype == np.float64:
                    self.__data = obj_double(context.context, data)
                elif USE_HALF and data.dtype == np.float16:
                    self.__data = obj_half(context.context, data)
                elif data.dtype == np.complex64:
                    self.__data = obj_float_complex(context.context, data)
                elif data.dtype == np.complex128:
                    self.__data = obj_double_complex(context.context, data)
                else:
                    raise TypeError("Data type not implemented")
                self.__dtype = data.dtype
                self.__shape = data.shape
            elif isinstance(data, obj_int):
                self.__data = data
                self.__dtype = np.int64
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif isinstance(data, obj_float):
                self.__data = data
                self.__dtype = np.float32
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif isinstance(data, obj_double):
                self.__data = data
                self.__dtype = np.float64
                self._shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif USE_HALF and isinstance(data, obj_half):
                self.__data = data
                self.__dtype = np.float16
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif isinstance(data, obj_float_complex):
                self.__data = data
                self.__dtype = np.complex64
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif isinstance(data, obj_double_complex):
                self.__data = data
                self.__dtype = np.complex128
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            elif isinstance(data, obj_uint16):
                self.__data = data
                self.__dtype = np.uint16
                self.__shape = tuple(data.shape[k] for k in range(len(data.shape)))
            else:
                raise TypeError("Data must be a list, a numpy array or a carmaWrap.obj")
            self.__size = self.__data.nbElem
        else:
            raise AttributeError("You must provide data or shape at least")
 

Member Function Documentation

__add__()

def naga.array.Array.__add__	(		self,
			idata
	)

Definition at line 133 of file array.py.

    def __add__(self, idata):
        tmp = self.copy()
        if isinstance(idata, Array):
            tmp.data.axpy(1, idata.data)
        elif isinstance(idata, np.ndarray):
            tmp.data.axpy(1, Array(idata).data)
        else:
            raise TypeError("operator + is defined only between Arrays and np.arrays")
        return tmp
 

Here is the call graph for this function:

__getitem__()

def naga.array.Array.__getitem__	(		self,
			indices
	)

Definition at line 161 of file array.py.

    def __getitem__(self, indices):
        return self.toarray().__getitem__(indices)
 

Here is the call graph for this function:

__mul__()

def naga.array.Array.__mul__	(		self,
			idata
	)

Definition at line 153 of file array.py.

    def __mul__(self, idata):
        if isinstance(idata, float) or isinstance(idata, int):
            tmp = self.copy()
            tmp.data.scale(idata)
            return tmp
        else:
            raise NotImplementedError("Operator not implemented yet")
 

Here is the call graph for this function:

__repr__()

def naga.array.Array.__repr__

(

self

)

Definition at line 130 of file array.py.

    def __repr__(self):
        return "GPU" + self.toarray().__repr__()
 

Here is the call graph for this function:

__sub__()

def naga.array.Array.__sub__	(		self,
			idata
	)

Definition at line 143 of file array.py.

    def __sub__(self, idata):
        tmp = self.copy()
        if isinstance(idata, Array):
            tmp.data.axpy(-1, idata.data)
        elif isinstance(idata, np.ndarray):
            tmp.data.axpy(-1, Array(idata).data)
        else:
            raise TypeError("operator + is defined only between Arrays and np.arrays")
        return tmp
 

Here is the call graph for this function:

argmax()

def naga.array.Array.argmax

(

self

)

Definition at line 205 of file array.py.

    def argmax(self):
        return self.data.aimax()
 

Here is the caller graph for this function:

argmin()

def naga.array.Array.argmin

(

self

)

Definition at line 211 of file array.py.

    def argmin(self):
        return self.data.aimin()
 

Here is the caller graph for this function:

copy()

def naga.array.Array.copy

(

self

)

Definition at line 164 of file array.py.

    def copy(self):
        tmp = Array(shape=self.shape, dtype=self.dtype)
        tmp.data.copy_from(self.data)
        return tmp
 

Here is the caller graph for this function:

dot()

def naga.array.Array.dot	(		self,
			idata
	)

Definition at line 169 of file array.py.

    def dot(self, idata):
        if isinstance(idata, np.ndarray):
            if idata.dtype == self.dtype:
                idata = Array(idata)
            else:
                raise TypeError("Data types must be the same for both arrays")
        if isinstance(idata, Array):
            if len(self.shape) == 1:
                if len(idata.shape) == 1:
                    if idata.shape == self.shape:
                        result = self.data.dot(idata.data, 1, 1)
                    else:
                        raise DimensionsError("Dimensions mismatch")
                elif len(idata.shape) == 2:
                    if idata.shape[0] == self.shape[0]:
                        result = Array(idata.data.gemv(self.data, op='T'))
                    else:
                        raise DimensionsError("Dimensions mismatch")
                else:
                    raise DimensionsError("Arrays must be 1D or 2D max")
            elif len(self.shape) == 2:
                if len(idata.shape) == 1:
                    if idata.shape[0] == self.shape[1]:
                        result = Array(self.data.gemv(idata.data))
                    else:
                        raise DimensionsError("Dimensions mismatch")
                elif len(idata.shape) == 2:
                    if self.shape[1] == idata.shape[0]:
                        result = Array(self.data.gemm(idata.data))
                    else:
                        raise DimensionsError("Dimensions mismatch")
            else:
                raise DimensionsError("Arrays must be 1D or 2D max")
 
        return result
 

max()

def naga.array.Array.max

(

self

)

Definition at line 208 of file array.py.

    def max(self):
        return self.toarray()[self.argmax()]
 

Here is the call graph for this function:

min()

def naga.array.Array.min

(

self

)

Definition at line 214 of file array.py.

    def min(self):
        return self.toarray()[self.argmin()]
 

Here is the call graph for this function:

sum()

def naga.array.Array.sum

(

self

)

Definition at line 217 of file array.py.

    def sum(self):
        return self.data.sum()
 

toarray()

def naga.array.Array.toarray

(

self

)

Definition at line 220 of file array.py.

    def toarray(self):
        tmp = np.array(self.data)
        return tmp
 
 

Here is the caller graph for this function:

Property Documentation

data

naga.array.Array.data = property(lambda x: x.__data)

static

Definition at line 128 of file array.py.

dtype

naga.array.Array.dtype = property(lambda x: x.__dtype)

static

Definition at line 127 of file array.py.

shape

naga.array.Array.shape = property(lambda x: x.__shape)

static

Definition at line 126 of file array.py.

---

The documentation for this class was generated from the following file:

• array.py