当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。