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


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


用法:

mxnet.ndarray.add(lhs, rhs)

参数

  • lhs(scalar or mxnet.ndarray.array) - 要添加的第一个数组。
  • rhs(scalar or mxnet.ndarray.array) - 要添加的第二个数组。如果lhs.shape != rhs.shape,它们必须可以广播到一个共同的形状。

返回

输入数组的元素总和。

返回类型

ND阵列

通过广播返回输入数组的元素总和。

等效于 lhs + rhsmx.nd.broadcast_add(lhs, rhs)mx.nd.broadcast_plus(lhs, rhs)

注意

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

例子

>>> x = mx.nd.ones((2,3))
>>> y = mx.nd.arange(2).reshape((2,1))
>>> z = mx.nd.arange(2).reshape((1,2))
>>> x.asnumpy()
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.]], dtype=float32)
>>> y.asnumpy()
array([[ 0.],
       [ 1.]], dtype=float32)
>>> z.asnumpy()
array([[ 0.,  1.]], dtype=float32)
>>> (x+2).asnumpy()
array([[ 3.,  3.,  3.],
       [ 3.,  3.,  3.]], dtype=float32)
>>> (x+y).asnumpy()
array([[ 1.,  1.,  1.],
       [ 2.,  2.,  2.]], dtype=float32)
>>> mx.nd.add(x,y).asnumpy()
array([[ 1.,  1.,  1.],
       [ 2.,  2.,  2.]], dtype=float32)
>>> (z + y).asnumpy()
array([[ 0.,  1.],
       [ 1.,  2.]], dtype=float32)

相关用法


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