15 lines
461 B
ReStructuredText
15 lines
461 B
ReStructuredText
mindspore.ops.Flatten
|
||
======================
|
||
|
||
.. py:class:: mindspore.ops.Flatten
|
||
|
||
扁平化（Flatten）输入Tensor，不改变0轴的size。
|
||
|
||
更多参考详见 :func:`mindspore.ops.flatten`。
|
||
|
||
输入：
|
||
- **input_x** (Tensor) - 待扁平化的Tensor，其shape为 :math:`(N, \ldots)`， :math:`N` 表示batch size。
|
||
|
||
输出：
|
||
Tensor，输出shape为 :math:`(N, X)` 的Tensor，其中 :math:`X` 是余下维度的乘积。
|