megengine.functional.broadcast_to¶

broadcast_to(inp, shape)[源代码]¶

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

参数

inp (Tensor) – 输入张量。
shape (Union[int, Iterable[int]]) – 目标形状。

返回类型

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)

megengine.functional.swapaxes

megengine.functional.expand_dims