mindspore.Tensor.flatten
- mindspore.Tensor.flatten(order='C')[源代码]
返回展开成一维的Tensor的副本。
- 参数:
order (str, 可选) - 可以在’C’和’F’之间进行选择。’C’表示按行优先(C风格)顺序展开。’F’表示按列优先顺序(Fortran风格)进行扁平化。仅支持’C’和’F’。默认值:’C’。
- 返回:
Tensor,具有与输入相同的数据类型。
- 异常:
TypeError - order 不是字符串类型。
ValueError - order 是字符串类型,但不是’C’或’F’。
- 参考:
mindspore.Tensor.reshape()
:在不改变数据的情况,改变Tensor的shape。mindspore.Tensor.ravel()
:返回一个连续扁平化的Tensor。- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> from mindspore import Tensor >>> x = Tensor(np.ones((2,3,4), dtype=np.float32)) >>> output = x.flatten() >>> print(output.shape) (24,)