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