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

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