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 - 如果输入的 shape 与 self 的形状不兼容,或者目标 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.]]