Python mxnet.ndarray.sparse.CSRNDArray用法及代碼示例

用法: class mxnet.ndarray.sparse.CSRNDArray(handle, writable=True)

基礎： mxnet.ndarray.sparse.BaseSparseNDArray

壓縮稀疏行格式的 2D NDArray 的稀疏表示。

CSRNDArray 將 NDArray 表示為三個單獨的數組：data、indptr 和 indices。它使用 CSR 表示，其中第 i 行的列索引存儲在 indices[indptr[i]:indptr[i+1]] 中，它們的相應值存儲在 data[indptr[i]:indptr[i+1]] 中。

給定行的列索引應按升序排序。不允許同一行的重複列條目。

示例：

屬性

`data`	CSRNDArray 的數據數組的深拷貝 NDArray。
`indices`	CSRNDArray 的索引數組的深拷貝 NDArray。
`indptr`	CSRNDArray 的 indptr 數組的深拷貝 NDArray。

>>> 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])

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.sparse.CSRNDArray。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。