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

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。