當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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


用法:

mxnet.ndarray.sparse.csr_matrix(arg1, shape=None, ctx=None, dtype=None)

參數

  • arg1(tuple of int, tuple of array_like, array_like, mxnet.ndarray.sparse.CSRNDArray, scipy.sparse.csr_matrix, scipy.sparse.coo_matrix, tuple of int or tuple of array_like) - 幫助實例化 csr 矩陣的參數。有關詳細信息,請參見上文。
  • shape(tuple of int, optional) - csr 矩陣的形狀。
  • ctx(mxnet.context.Context, optional) - 設備上下文(默認為當前默認上下文)。
  • dtype(str or numpy.dtype, optional) - 輸出數組的數據類型。

返回

具有 csr 存儲表示的 CSRNDArray

返回類型

mxnet.ndarray.sparse.CSRNDArray

創建 CSRNDArray ,一個具有壓縮稀疏行 (CSR) 格式的二維數組。

CSRNDArray 可以通過多種方式實例化:

  • csr_matrix(D):
    用密集的二維數組構造一個 CSRNDArray D
    • D(array_like) - 一個暴露數組接口的對象,一個對象__array__方法返回一個數組或任何(嵌套)序列。
    • ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。
    • 類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型是D.dtype如果D是 NDArray 或 numpy.ndarray float32 否則。
  • csr_matrix(S)
    構造一個帶有稀疏二維數組的 CSRNDArray S
    • S(CSRNDArray or scipy.sparse.csr.csr_matrix) - 一個稀疏矩陣。
    • ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。
    • 類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型是S.dtype.
  • csr_matrix((M, N))
    構造一個形狀為 (M, N) 的空 CSRNDArray
    • M(int) - 矩陣中的行數
    • N(int) - 矩陣中的列數
    • ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。
    • 類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型為 float32。
  • csr_matrix((數據、索引、indptr))
    使用三個單獨的數組,根據壓縮稀疏行格式的定義構造一個 CSRNDArray,其中第 i 行的列索引存儲在indices[indptr[i]:indptr[i+1]]並且它們對應的值存儲在data[indptr[i]:indptr[i+1]].給定行的列索引預計為按升序排列。不允許同一行的重複列條目。
    • 數據(array_like) - 一個暴露數組接口的對象,它以行優先順序保存矩陣的所有非零條目。
    • index (array_like) - 一個暴露數組接口的對象,它將每個非零元素的列索引存儲在data.
    • 索引(array_like) - 一個暴露數組接口的對象,它將偏移存儲到data矩陣的每一行的第一個非零元素數。
    • 形狀(tuple of int, optional) - 數組的形狀。默認形狀是從索引和 indptr 數組中推斷出來的。
    • ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。
    • 類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型是data.dtype如果data是 NDArray 或 numpy.ndarray float32 否則。
  • csr_matrix((數據,(行,列)))
    使用三個單獨的數組構造基於 COOrdinate 格式的 CSRNDArray,其中 row[i] 是元素的行索引,col[i] 是元素的列索引,data[i] 是元素對應的數據。輸入中所有缺失的元素都被視為零。
    • 數據(array_like) - 一個暴露數組接口的對象,它以 COO 格式保存矩陣的所有非零條目。
    • (array_like) - 一個暴露數組接口的對象,它將每個非零元素的行索引存儲在data.
    • 山口(array_like) - 一個暴露數組接口的對象,它將每個非零元素的 col 索引存儲在data.
    • 形狀(tuple of int, optional) - 數組的形狀。默認形狀是從rowcol數組。
    • ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。
    • 類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型為 float32。

示例

>>> a = mx.nd.sparse.csr_matrix(([1, 2, 3], [1, 0, 2], [0, 1, 2, 2, 3]), shape=(4, 3))
>>> a.asnumpy()
array([[ 0.,  1.,  0.],
       [ 2.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  3.]], dtype=float32)

相關用法


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