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

用法:
mxnet.ndarray.broadcast_axis(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_axis。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：