broadcast_to

paddle. broadcast_to ( x, shape, name=None ) [源代码]

根据 shape 指定的形状广播 x,广播后,x 的形状和 shape 指定的形状一致。

x 的维数和 shape 的元素数应小于等于 6,并且 shape 中的元素数应该大于等于 x 的维数。扩展的维度的维度值应该为 1。

参数

  • x (Tensor) - 输入的 Tensor,数据类型为:bool、float16、float32、float64、int32、int64、uint8 或 uint16。

  • shape (tuple|list|Tensor) - 给定输入 x 扩展后的形状,若 shape 为 list 或者 tuple,则其中的元素值应该为整数或者是形状为 1-D 或 0-D 的 Tensor,若 shape 类型为 Tensor,则其应该为 1-D Tensor。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor,数据类型与 x 相同。

代码示例

>>> import paddle

>>> data = paddle.to_tensor([1, 2, 3], dtype='int32')
>>> out = paddle.broadcast_to(data, shape=[2, 3])
>>> print(out)
Tensor(shape=[2, 3], dtype=int32, place=Place(cpu), stop_gradient=True,
[[1, 2, 3],
 [1, 2, 3]])