mindspore.ops.BroadcastTo
- class mindspore.ops.BroadcastTo(shape)[源代码]
将输入shape广播到目标shape。
更多细节请参考
mindspore.ops.broadcast_to()
。- 参数:
shape (tuple) - 指定广播的目标shape。
- 输入:
input_x (Tensor) - BroadcastTo输入,是任意维度的Tensor。
- 输出:
Tensor,与目标 shape 相同,数据类型与 input_x 相同。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> shape = (2, 3) >>> x = Tensor(np.array([1, 2, 3]).astype(np.float32)) >>> output = ops.BroadcastTo(shape=shape)(x) >>> print(output) [[1. 2. 3.] [1. 2. 3.]] >>> >>> shape = (-1, 2) >>> x = Tensor(np.array([[1], [2]]).astype(np.float32)) >>> output = ops.BroadcastTo(shape=shape)(x) >>> print(output) [[1. 1.] [2. 2.]]