代码拉取完成,页面将自动刷新
.. py:function:: mindspore.mint.flatten(input, start_dim=0, end_dim=-1) 沿着从 `start_dim` 到 `end_dim` 的维度,对输入Tensor进行展平。 参数: - **input** (Tensor) - 输入Tensor。 - **start_dim** (int, 可选) - 要展平的第一个维度。默认值: ``0`` 。 - **end_dim** (int, 可选) - 要展平的最后一个维度。默认值: ``-1`` 。 返回: Tensor。如果没有维度被展平,返回原始的输入 `input`,否则返回展平后的Tensor。如果 `input` 是零维Tensor,将会返回一个一维Tensor。 异常: - **TypeError** - `input` 不是Tensor。 - **TypeError** - `start_dim` 或 `end_dim` 不是int类型。 - **ValueError** - 规范化后, `start_dim` 大于 `end_dim` 。 - **ValueError** - `start_dim` 或 `end_dim` 不在 [-input.dim, input.dim-1] 范围内。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。