當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。