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

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


用法:

mxnet.ndarray.op.broadcast_to(data=None, shape=_Null, out=None, name=None, **kwargs)

參數：

  • data：(NDArray) - 輸入
  • shape：(Shape(tuple), optional, default=[]) - 所需數組的形狀。如果 dim 與原始值相同，我們可以將其設置為零。例如A = broadcast_to(B, shape=(10, 0, 0))具有相同的含義A = broadcast_axis(B, axis=0, size=10).
  • out：(NDArray, optional) - 輸出 NDArray 來保存結果。

返回：

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

將輸入數組廣播到新形狀。

廣播是一種機製，它允許 NDArrays 有效地對不同形狀的數組執行算術運算，而無需創建數組的多個副本。另請參閱Broadcasting 以獲得更多解釋。

在大小為 1 的軸上允許廣播，例如從 (2,1,3,1) 到 (2,8,3,9) 。元素將在廣播軸上複製。

例如：

broadcast_to([[1,2,3]], shape=(2,3)) = [[ 1.,  2.,  3.],
                                        [ 1.,  2.,  3.]])

您不想更改的維度也可以保留為0，即複製原始值。因此，使用 shape=(2,0) ，我們將獲得與上述示例相同的結果。

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.op.broadcast_to。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。