Python mxnet.ndarray.choose_element_0index用法及代码示例

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

用法:

参数：

返回：

返回类型：