forked from huawei/mindspore2022
36 lines
1.1 KiB
ReStructuredText
36 lines
1.1 KiB
ReStructuredText
mindspore.nn.Flatten
|
||
====================
|
||
|
||
.. py:class:: mindspore.nn.Flatten
|
||
|
||
对输入Tensor的第0维之外的维度进行展平操作。
|
||
|
||
**输入：**
|
||
|
||
- **x** (Tensor) - 要展平的输入Tensor。shape为 :math:`(N, *)`，其中 :math:`*` 表示任意的附加维度。数据类型为 `number <https://www.mindspore.cn/docs/api/zh-CN/master/api_python/mindspore.html#mindspore.dtype>`_。
|
||
|
||
**输出：**
|
||
|
||
Tensor，shape为 :math:`(N, X)`，其中 :math:`X` 是输入 `x` 的shape除N之外的其余维度的乘积。
|
||
|
||
**异常：**
|
||
|
||
**TypeError** - `x` 不是Tensor。
|
||
|
||
**支持平台：**
|
||
|
||
``Ascend`` ``GPU`` ``CPU``
|
||
|
||
**样例：**
|
||
|
||
>>> x = Tensor(np.array([[[1.2, 1.2], [2.1, 2.1]], [[2.2, 2.2], [3.2, 3.2]]]), mindspore.float32)
|
||
>>> net = nn.Flatten()
|
||
>>> output = net(x)
|
||
>>> print(output)
|
||
[[1.2 1.2 2.1 2.1]
|
||
[2.2 2.2 3.2 3.2]]
|
||
>>> print(f"Before flatten the x shape is {x.shape}.")
|
||
Before flatten the x shape is (2, 2, 2)
|
||
>>> print(f"After flatten the output shape is {output.shape}.")
|
||
After flatten the output shape is (2, 4)
|
||
|