CSRNDArray¶

class mxnet.ndarray.sparse.CSRNDArray(handle, writable=True)[source]¶

A sparse representation of 2D NDArray in the Compressed Sparse Row format.

A CSRNDArray represents an NDArray as three separate arrays: data, indptr and indices. It uses the CSR representation where the column indices for row i are stored in indices[indptr[i]:indptr[i+1]] and their corresponding values are stored in data[indptr[i]:indptr[i+1]].

The column indices for a given row are expected to be sorted in ascending order. Duplicate column entries for the same row are not allowed.

Example

>>> a = mx.nd.array([[0, 1, 0], [2, 0, 0], [0, 0, 0], [0, 0, 3]])
>>> a = a.tostype('csr')
>>> a.data.asnumpy()
array([ 1.,  2.,  3.], dtype=float32)
>>> a.indices.asnumpy()
array([1, 0, 2])
>>> a.indptr.asnumpy()
array([0, 1, 2, 2, 3])

Array attributes¶

`CSRNDArray.context`	Device context of the array.
`CSRNDArray.data`	A deep copy NDArray of the data array of the CSRNDArray.
`CSRNDArray.dtype`	Data-type of the array’s elements.
`CSRNDArray.indices`	A deep copy NDArray of the indices array of the CSRNDArray.
`CSRNDArray.indptr`	A deep copy NDArray of the indptr array of the CSRNDArray.
`CSRNDArray.shape`	Tuple of array dimensions.
`CSRNDArray.stype`	Storage-type of the array.

Array creation¶

CSRNDArray.zeros_like

Convenience fluent method for zeros_like().

Manipulation¶

Array conversion¶

`CSRNDArray.as_in_context`	Returns an array on the target device with the same value as this array.
`CSRNDArray.asnumpy`	Return a dense `numpy.ndarray` object with value copied from this array
`CSRNDArray.asscalar`	Returns a scalar whose value is copied from this array.
`CSRNDArray.asscipy`	Returns a `scipy.sparse.csr.csr_matrix` object with value copied from this array
`CSRNDArray.astype`	Return a copy of the array after casting to a specified type.
`CSRNDArray.copy`	Makes a copy of this `NDArray`, keeping the same context.
`CSRNDArray.copyto`	Copies the value of this array to another array.
`CSRNDArray.tostype`	Return a copy of the array with chosen storage type.

Array inspection¶

CSRNDArray.check_format

Check whether the NDArray format is valid.

Array reduction¶

`CSRNDArray.mean`	Convenience fluent method for `mean()`.
`CSRNDArray.norm`	Convenience fluent method for `norm()`.
`CSRNDArray.sum`	Convenience fluent method for `sum()`.

Indexing¶

`CSRNDArray.__getitem__`	x.__getitem__(i) <=> x[i]
`CSRNDArray.__setitem__`	x.__setitem__(i, y) <=> x[i]=y
`CSRNDArray.slice`	Convenience fluent method for `slice()`.

Joining arrays¶

concat

Joins input arrays along a given axis.

Lazy evaluation¶

CSRNDArray.wait_to_read

Waits until all previous write operations on the current array are finished.

Math¶

Array rounding¶

`CSRNDArray.ceil`	Convenience fluent method for `ceil()`.
`CSRNDArray.fix`	Convenience fluent method for `fix()`.
`CSRNDArray.floor`	Convenience fluent method for `floor()`.
`CSRNDArray.round`	Convenience fluent method for `round()`.
`CSRNDArray.rint`	Convenience fluent method for `rint()`.
`CSRNDArray.trunc`	Convenience fluent method for `trunc()`.

Trigonometric functions¶

`CSRNDArray.arcsin`	Convenience fluent method for `arcsin()`.
`CSRNDArray.arctan`	Convenience fluent method for `arctan()`.
`CSRNDArray.degrees`	Convenience fluent method for `degrees()`.
`CSRNDArray.radians`	Convenience fluent method for `radians()`.
`CSRNDArray.sin`	Convenience fluent method for `sin()`.
`CSRNDArray.tan`	Convenience fluent method for `tan()`.

Hyperbolic functions¶

`CSRNDArray.arcsinh`	Convenience fluent method for `arcsinh()`.
`CSRNDArray.arctanh`	Convenience fluent method for `arctanh()`.
`CSRNDArray.sinh`	Convenience fluent method for `sinh()`.
`CSRNDArray.tanh`	Convenience fluent method for `tanh()`.

Exponents and logarithms¶

`CSRNDArray.expm1`	Convenience fluent method for `expm1()`.
`CSRNDArray.log1p`	Convenience fluent method for `log1p()`.

Powers¶

`CSRNDArray.sqrt`	Convenience fluent method for `sqrt()`.
`CSRNDArray.square`	Convenience fluent method for `square()`.

Miscellaneous¶

`CSRNDArray.abs`	Convenience fluent method for `abs()`.
`CSRNDArray.clip`	Convenience fluent method for `clip()`.
`CSRNDArray.sign`	Convenience fluent method for `sign()`.