当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法:

mxnet.symbol.contrib.index_array(data=None, axes=_Null, name=None, attr=None, out=None, **kwargs)

参数

  • data(Symbol) - 输入数据
  • axes(Shape or None, optional, default=None) - 要包含在索引数组中的轴。支持负值。
  • name(string, optional.) - 结果符号的名称。

返回

结果符号。

返回类型

Symbol

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

对于形状为 的输入数组,index_array 返回一个 数组 idx ,其中

此外,当指定参数 axes 时,idx 将是一个 数组,其中 maxes 的长度,并且以下等式将成立:

例子:

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.symbol.contrib.index_array。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。