Python mxnet.ndarray.sparse.retain用法及代码示例

用法:
mxnet.ndarray.sparse.retain(data=None, indices=None, out=None, name=None, **kwargs)

参数：

data：(NDArray) - sparse_retain 运算符的输入数组。
indices：(NDArray) - 将保留的行 ID 的索引数组。
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

从行稀疏矩阵中选择用户输入索引数组指定的行并将它们保存在输出稀疏矩阵中。

例子：

data = [[1, 2], [3, 4], [5, 6]]
indices = [0, 1, 3]
shape = (4, 2)
rsp_in = row_sparse_array(data, indices)
to_retain = [0, 3]
rsp_out = retain(rsp_in, to_retain)
rsp_out.data = [[1, 2], [5, 6]]
rsp_out.indices = [0, 3]

retain 输出的存储类型取决于输入的存储类型

保留(row_sparse，默认)=row_sparse
否则，不支持retain

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.sparse.retain。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：