megengine.functional.reshape¶

reshape(inp, target_shape)[源代码]¶

将一个张量重塑为给定的目标形状；逻辑元素的总数必须保持不变

参数

inp (Tensor) – 输入张量。
target_shape (Iterable[int]) – 目标形状，所有组件将被连接成目标形状，其中可能包含用来表示unspec_axis的值为-1的元素。

实际案例

import numpy as np
from megengine import tensor
import megengine.functional as F
x = tensor(np.arange(12, dtype=np.int32))
out = F.reshape(x, (3, 4))
print(out.numpy())

输出：

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

返回类型: Tensor

megengine.functional.copy

megengine.functional.flatten