Python mxnet.ndarray.op.pick用法及代碼示例

用法:
mxnet.ndarray.op.pick(data=None, index=None, axis=_Null, keepdims=_Null, mode=_Null, out=None, name=None, **kwargs)

參數：

data：(NDArray) - 輸入數組
index：(NDArray) - 索引數組
axis：(int or None, optional, default='-1') - int 或無。拾取元素的軸。負值表示從右到左索引。如果是None，索引中的元素 w.r.t 將被選中。
keepdims：(boolean, optional, default=0) - 如果為真，我們選擇元素的軸將作為尺寸為 1 的維度留在結果中。
mode：({'clip', 'wrap'},optional, default='clip') - 指定越界索引的行為方式。默認為“clip”。 “clip” 表示剪輯到範圍。因此，如果提到的所有索引都太大，則它們將替換為指向軸上最後一個元素的索引。 “wrap” 表示環繞。
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

根據沿給定軸的輸入索引從輸入數組中挑選元素。

給定一個形狀為 (d0, d1) 的輸入數組和形狀為 (i0,) 的索引，結果將是一個形狀為 (i0,) 的輸出數組，其中：

output[i] = input[i, indices[i]]

默認情況下，如果提及的任何索引太大，則將其替換為沿軸尋址最後一個元素的索引(clip 模式)。

此函數支持 n 維輸入和 (n-1) 維索引數組。

例子：

x = [[ 1.,  2.],
     [ 3.,  4.],
     [ 5.,  6.]]

// picks elements with specified indices along axis 0
pick(x, y=[0,1], 0) = [ 1.,  4.]

// picks elements with specified indices along axis 1
pick(x, y=[0,1,0], 1) = [ 1.,  4.,  5.]

// picks elements with specified indices along axis 1 using 'wrap' mode
// to place indicies that would normally be out of bounds
pick(x, y=[2,-1,-2], 1, mode='wrap') = [ 1.,  4.,  5.]

y = [[ 1.],
     [ 0.],
     [ 2.]]

// picks elements with specified indices along axis 1 and dims are maintained
pick(x, y, 1, keepdims=True) = [[ 2.],
                               [ 3.],
                               [ 6.]]

相關用法

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

用法:

參數：

返回：

返回類型：