Python mxnet.ndarray.contrib.index_array用法及代码示例

用法:
mxnet.ndarray.contrib.index_array(data=None, axes=_Null, out=None, name=None, **kwargs)

参数：

data：(NDArray) - 输入数据
axes：(Shape or None, optional, default=None) - 要包含在索引数组中的轴。支持负值。
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

返回输入数组的索引数组。

对于形状为 \((d_1, d_2, ..., d_n)\) 的输入数组，index_array 返回一个 \((d_1, d_2, ..., d_n, n)\) 数组 idx ，其中 \(idx[i_1, i_2, ..., i_n, :] = [i_1, i_2, ..., i_n]\) 。

此外，当指定参数 axes 时，idx 将是一个 \((d_1, d_2, ..., d_n, m)\) 数组，其中 m 是 axes 的长度，并且以下等式将成立：\(idx[i_1, i_2, ..., i_n, j] = i_{axes[j]}\) 。

例子：

x = mx.nd.ones((3, 2))

mx.nd.contrib.index_array(x) = [[[0 0]
                                 [0 1]]

                                [[1 0]
                                 [1 1]]

                                [[2 0]
                                 [2 1]]]

x = mx.nd.ones((3, 2, 2))

mx.nd.contrib.index_array(x, axes=(1, 0)) = [[[[0 0]
                                               [0 0]]

                                              [[1 0]
                                               [1 0]]]


                                             [[[0 1]
                                               [0 1]]

                                              [[1 1]
                                               [1 1]]]


                                             [[[0 2]
                                               [0 2]]

                                              [[1 2]
                                               [1 2]]]]

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.contrib.index_array。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：