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


Python mxnet.symbol.op.broadcast_axes用法及代碼示例


用法:

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

參數

  • data(Symbol) - 輸入
  • axis(Shape(tuple), optional, default=[]) - 執行廣播的軸。
  • size(Shape(tuple), optional, default=[]) - 廣播軸的目標大小。
  • name(string, optional.) - 結果符號的名稱。

返回

結果符號。

返回類型

Symbol

在特定軸上廣播輸入數組。

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