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,)