当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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