megengine.functional.broadcast_to

broadcast_to(inp, shape)[源代码]

将张量广播至指定的形状。

参数
返回类型

Tensor

返回

输出张量。

实际案例

>>> import numpy as np
>>> data = Tensor(np.arange(0, 3, dtype=np.float32).reshape(3))
>>> out = F.broadcast_to(data, (2, 3))
>>> out.numpy()
array([[0., 1., 2.],
       [0., 1., 2.]], dtype=float32)