megengine.functional.reshape¶

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

在不改变数据的情况下更改 Tensor 的形状。

参数

inp (Tensor) – 需要被更改形状的输入 Tensor
target_shape (Iterable[int]) – 目标形状与原始形状兼容。一个外形尺寸允许为`-1`。当形状尺寸为“-1”时，必须根据输入 Tensor 的长度和剩余尺寸推断出相应的输出 Tensor 的形状尺寸。

返回类型

Tensor

返回

输出 Tensor 和输入 Tensor 拥有相同的数据类型。

实际案例

>>> x = F.arange(12)
>>> x
Tensor([ 0.  1.  2.  3.  4.  5.  6.  7.  8.  9. 10. 11.], device=xpux:0)
>>> F.reshape(x, (3, 4))
Tensor([[ 0.  1.  2.  3.]
 [ 4.  5.  6.  7.]
 [ 8.  9. 10. 11.]], device=xpux:0)
>>> F.reshape(x, (2, -1))
Tensor([[ 0.  1.  2.  3.  4.  5.]
 [ 6.  7.  8.  9. 10. 11.]], device=xpux:0)

megengine.functional.copy

megengine.functional.flatten