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


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


用法:

mxnet.ndarray.power(base, exp)

参数

  • base(scalar or NDArray) - 基本数组
  • exp(scalar or NDArray) - index 数组。如果base.shape != exp.shape,它们必须可以广播到一个共同的形状。

返回

x 中的基数增加到 y 中的 index 。

返回类型

ND阵列

返回第一个数组元素从第二个数组的幂次方的结果,按元素进行广播。

等效于 base ** expmx.nd.broadcast_power(lhs, rhs)

注意

如果两个数组的对应维度具有相同的大小或其中一个的大小为 1,则这些数组可以广播为一个共同的形状。

例子

>>> x = mx.nd.ones((2,3))*2
>>> y = mx.nd.arange(1,3).reshape((2,1))
>>> z = mx.nd.arange(1,3).reshape((2,1))
>>> x.asnumpy()
array([[ 2.,  2.,  2.],
       [ 2.,  2.,  2.]], dtype=float32)
>>> y.asnumpy()
array([[ 1.],
       [ 2.]], dtype=float32)
>>> z.asnumpy()
array([[ 1.],
       [ 2.]], dtype=float32)
>>> (x**2).asnumpy()
array([[ 4.,  4.,  4.],
       [ 4.,  4.,  4.]], dtype=float32)
>>> (x**y).asnumpy()
array([[ 2.,  2.,  2.],
       [ 4.,  4.,  4.]], dtype=float32)
>>> mx.nd.power(x,y).asnumpy()
array([[ 2.,  2.,  2.],
       [ 4.,  4.,  4.]], dtype=float32)
>>> (z**y).asnumpy()
array([[ 1.],
       [ 4.]], dtype=float32)

相关用法


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