mindspore.Tensor.flatten

查看源文件
mindspore.Tensor.flatten(start_dim=0, end_dim=- 1)

沿着从 start_dimend_dim 的维度,对输入Tensor进行展平。

参数:
  • start_dim (int, 可选) - 要展平的第一个维度。默认值: 0

  • end_dim (int, 可选) - 要展平的最后一个维度。默认值: -1

返回:

Tensor。如果没有维度被展平,返回原始的输入 self,否则返回展平后的Tensor。如果 self 是零维Tensor,将会返回一个一维Tensor。

异常:
  • TypeError - start_dimend_dim 不是int类型。

  • ValueError - 规范化后, start_dim 大于 end_dim

  • ValueError - start_dimend_dim 不在 [-self.dim, self.dim-1] 范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.ones(shape=[1, 2, 3, 4]), mindspore.float32)
>>> output = input_x.flatten(0, -1)
>>> print(output.shape)
(24,)
mindspore.Tensor.flatten(order='C', *, start_dim=0, end_dim=- 1)

沿着从 start_dimend_dim 的维度,对输入Tensor进行展平。

参数:
  • order (str, 可选) - 仅支持 'C''F''C' 表示按行优先顺序 (C风格) 展平, 'F' 表示按列优先顺序 (Fortran风格) 展平。默认值: 'C'

关键字参数:
  • start_dim (int, 可选) - 要展平的第一个维度。默认值: 0

  • end_dim (int, 可选) - 要展平的最后一个维度。默认值: -1

返回:

Tensor。如果没有维度被展平,返回原始的输入 self,否则返回展平后的Tensor。如果 self 是零维Tensor,将会返回一个一维Tensor。

异常:
  • TypeError - order 不是string类型。

  • ValueError - order 是string类型,但不是 'C''F'

  • TypeError - start_dimend_dim 不是int类型。

  • ValueError - 规范化后, start_dim 大于 end_dim

  • ValueError - start_dimend_dim 不在 [-self.dim, self.dim-1] 范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.ones(shape=[1, 2, 3, 4]), mindspore.float32)
>>> output = input_x.flatten(order='C')
>>> print(output.shape)
(24,)