megengine.functional.broadcast_to

broadcast_to(inp, shape)

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

参数

• inp (Tensor) – 输入张量。

• shape (Union[int, Iterable[int]]) – 目标形状。

返回类型

Tensor

返回

输出张量。

实际案例

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.]]

上一页

megengine.functional.transpose

下一页

megengine.functional.expand_dims