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


Python mxnet.ndarray.broadcast_to用法及代码示例


用法:

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