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

Python mxnet.ndarray.op.broadcast_axis用法及代碼示例


用法:

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