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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。