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


Python mxnet.ndarray.NDArray.broadcast_like用法及代碼示例


用法:

broadcast_like(other)

參數

other(NDArray) - 具有所需陣列形狀的陣列。

返回

具有所需形狀的 NDArray 不與此數組共享數據,即使新形狀與 self.shape 相同。

返回類型

ND陣列

將輸入數組廣播到其他的形狀。

隻允許在大小為 1 的軸上廣播。新形狀不能更改維度數。例如,您可以從形狀 (2, 1) 廣播到 (2, 3),但不能從形狀 (2, 3) 廣播到 (2, 3, 3)。

例子

>>> x = mx.nd.arange(0,3).reshape((1,3,1))
>>> x.asnumpy()
array([[[ 0.],
        [ 1.],
        [ 2.]]], dtype=float32)
>>> y = x.broadcast_like(mx.nd.ones((2,3,3)))
>>> y.asnumpy()
array([[[ 0.,  0.,  0.],
        [ 1.,  1.,  1.],
        [ 2.,  2.,  2.]],

       [[ 0.,  0.,  0.],
        [ 1.,  1.,  1.],
        [ 2.,  2.,  2.]]], dtype=float32)

相關用法


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