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

用法:

參數：

返回：

返回類型：