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

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


用法:

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

參數：

  • arg1：(NDArray, numpy.ndarray, mxnet.ndarray.sparse.RowSparseNDArray, tuple of int or tuple of array_like) - 幫助實例化行稀疏 ndarray 的參數。有關詳細信息，請參見上文。
  • shape：(tuple of int, optional) - 行稀疏 ndarray 的形狀。 (默認值 = 無)
  • ctx：(mxnet.context.Context, optional) - 設備上下文(默認為當前默認上下文)。
  • dtype：(str or numpy.dtype, optional) - 輸出數組的數據類型。 (默認值 = 無)

返回：

具有 row_sparse 存儲表示的 RowSparseNDArray。

返回類型：

mxnet.ndarray.sparse.RowSparseNDArray

創建一個 RowSparseNDArray ，一個多維行稀疏數組，在給定索引處具有一組張量切片。

RowSparseNDArray 可以通過多種方式實例化：

  • row_sparse_array(D):

    用密集的ndarray構造一個RowSparseNDArrayD-D(array_like) - 一個暴露數組接口的對象，一個對象__array__方法返回一個數組或任何(嵌套)序列。 -ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。 -類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型是D.dtype如果D是 NDArray 或 numpy.ndarray float32 否則。

  • row_sparse_array(S)

    用稀疏的 ndarray 構造 RowSparseNDArrayS-S(RowSparseNDArray) - 一個稀疏的 ndarray。 -ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。 -類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型是S.dtype.

  • row_sparse_array((D0, D1 .. Dn))

    用形狀構造一個空的RowSparseNDArray(D0, D1, ... Dn)-D0, D1 .. DN(int) - ndarray 的形狀 -ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。 -類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型為 float32。

  • row_sparse_array((數據，索引))

    使用兩個單獨的數組，根據行稀疏格式的定義構造一個RowSparseNDArray，其中indices存儲具有非零的行切片的索引，而值存儲在data.對應的NDArraydense由RowSparseNDArray 表示rsp已dense[rsp.indices[i], :, :, :, ...] = rsp.data[i, :, :, :, ...]的行索引預計為按升序排列。-數據(array_like) - 一個暴露數組接口的對象，它包含數組的所有非零行切片。 - index (array_like) - 一個暴露數組接口的對象，它存儲每個行切片的非零元素的行索引。 -形狀(tuple of int, optional) - 數組的形狀。默認形狀是從索引和 indptr 數組中推斷出來的。 -ctx(Context, optional) - 設備上下文(默認為當前默認上下文)。 -類型(str or numpy.dtype, optional) - 輸出數組的數據類型。默認數據類型為 float32。

例子：

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

相關用法


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