mindspore.Tensor.broadcast_to

mindspore.Tensor.broadcast_to(shape)[源代码]

将输入shape广播到目标shape。

更多细节请参考 mindspore.ops.broadcast_to()

参数:
  • shape (tuple) - 要广播的目标形状。可以由用户指定,或在要广播的维度上指定-1,它将被该位置的输入张量形状替换。

返回:

Tensor,形状为用户指定的 shape,类型和 self 相同。

异常:
  • TypeError - 如果输入的 shape 参数不是tuple类型。

  • ValueError - 如果输入的 shapeself 的形状不兼容,或者目标 shape 中的-1位于无效位置。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> shape = (2, 3)
>>> x = Tensor(np.array([1, 2, 3]).astype(np.float32))
>>> output = x.broadcast_to(shape)
>>> print(output)
[[1. 2. 3.]
 [1. 2. 3.]]
>>> shape = (-1, 2)
>>> x = Tensor(np.array([[1], [2]]).astype(np.float32))
>>> output = x.broadcast_to(shape)
>>> print(output)
[[1. 1.]
 [2. 2.]]