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

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


用法:

mxnet.symbol.sparse.retain(data=None, indices=None, name=None, attr=None, out=None, **kwargs)

参数：

  • data：(Symbol) - sparse_retain 运算符的输入数组。
  • indices：(Symbol) - 将保留的行 ID 的索引数组。
  • name：(string, optional.) - 结果符号的名称。

返回：

结果符号。

返回类型：

Symbol

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

例子：

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