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


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


用法:

mxnet.ndarray.broadcast_axes(data=None, axis=_Null, size=_Null, out=None, name=None, **kwargs)

参数

  • data(NDArray) - 输入
  • axis(Shape(tuple), optional, default=[]) - 执行广播的轴。
  • size(Shape(tuple), optional, default=[]) - 广播轴的目标大小。
  • out(NDArray, optional) - 输出 NDArray 来保存结果。

返回

out- 此函数的输出。

返回类型

NDArray 或 NDArray 列表

在特定轴上广播输入数组。

在大小为 1 的轴上允许广播,例如从 (2,1,3,1)(2,8,3,9) 。元素将在广播轴上复制。

broadcast_axes 是函数 broadcast_axis 的别名。

例子:

// given x of shape (1,2,1)
x = [[[ 1.],
      [ 2.]]]

// broadcast x on on axis 2
broadcast_axis(x, axis=2, size=3) = [[[ 1.,  1.,  1.],
                                      [ 2.,  2.,  2.]]]
// broadcast x on on axes 0 and 2
broadcast_axis(x, axis=(0,2), size=(2,3)) = [[[ 1.,  1.,  1.],
                                              [ 2.,  2.,  2.]],
                                             [[ 1.,  1.,  1.],
                                              [ 2.,  2.,  2.]]]

相关用法


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