megengine.functional.broadcast_to

broadcast_to(inp, shape)[源代码]

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

参数
  • inp – 输入张量。

  • shape – 目标形状。

返回

输出张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

data = tensor(np.arange(0, 3, dtype=np.float32).reshape(3))
out = F.broadcast_to(data, (2, 3))
print(out.numpy())

输出:

[[0. 1. 2.]
 [0. 1. 2.]]