Python mxnet.ndarray.contrib.count_sketch用法及代碼示例

用法:
mxnet.ndarray.contrib.count_sketch(data=None, h=None, s=None, out_dim=_Null, processing_batch_size=_Null, out=None, name=None, **kwargs)

參數：

data：(NDArray) - 將數據輸入到 CountSketchOp。
h：(NDArray) - 索引向量
s：(NDArray) - 符號向量
out_dim：(int, required) - 輸出維度。
processing_batch_size：(int, optional, default='32') - 一次處理多少個草圖向量。
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

將CountSketch應用於輸入：將d-dimension數據映射到k-dimension數據”

注意：

count_sketch 僅在 GPU 上可用。

假設輸入數據的形狀為(N, d)，符號哈希表s的形狀為(N, d)，索引哈希表h的形狀為(N, d)，映射維度out_dim=k，s中的每個元素都是+ 1 或 -1，h 中的每個元素都是從 0 到 k-1 的隨機整數。然後算子計算：

\[out[h[i]] += data[i] * s[i]\]

例子：

out_dim = 5
x = [[1.2, 2.5, 3.4],[3.2, 5.7, 6.6]]
h = [[0, 3, 4]]
s = [[1, -1, 1]]
mx.contrib.ndarray.count_sketch(data=x, h=h, s=s, out_dim = 5) = [[1.2, 0, 0, -2.5, 3.4],
                                                                  [3.2, 0, 0, -5.7, 6.6]]

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.contrib.count_sketch。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

注意：