mxnet基础到提高(47)-RowSparseNDArray

类mxnet.ndarray.sparse.RowSparseNDArray(handle,writable = True)[
bases:mxnet.ndarray.sparse.BaseSparseNDArray
在给定索引处的一组NDArray行切片的稀疏表示。
RowSparseNDArray使用两个单独的数组表示多维NDArray:数据和索引。 尺寸数必须至少为2。

数据:形状为[D0,D1,…,Dn]的任何dtype的NDArray。

索引:形状为[D0]的一维int64 NDArray,其值按升序排序。

属性

data
RowSparseNDArray的数据数组的深拷贝NDArray。

indices
RowSparseNDArray的索引数组的深层副本NDArray。

索引(indices )存储具有非零值的行切片的索引,而值存储在数据(data)中。
由RowSparse 的NDArray b表示的相应NDArray x为

x[rsp.indices[i], :, :, :, ...] = rsp.data[i, :, :, :, ...]

RowSparseNDArray通常用于表示形状为[LARGE0,D1,… Dn]的大型NDArray的非零行切片,其中LARGE0 >> D0且大多数行切片为零。
RowSparseNDArray主要用于梯度的定义中,以用于具有稀疏梯度的操作(例如,稀疏点和稀疏嵌入)。

上一篇:leetcode-47


下一篇:leetcode-剑指47-OK