mindspore/docs/api/api_python/ops/mindspore.ops.MaskedSelect.rst

20 lines
656 B
ReStructuredText
Raw Normal View History

mindspore.ops.MaskedSelect
===========================
.. py:class:: mindspore.ops.MaskedSelect
2022-05-25 14:16:31 +08:00
返回一个一维张量,其中的内容是 `x` 张量中对应于 `mask` 张量中True位置的值。`mask` 的shape与 `x` 的shape不需要一样但必须符合广播规则。
**输入:**
2022-05-25 14:16:31 +08:00
- **x** (Tensor) - 它的shape是 :math:`(x_1, x_2, ..., x_R)`
- **mask** (Tensor[bool]) - 它的shape是 :math:`(x_1, x_2, ..., x_R)`
**输出:**
2022-05-25 14:16:31 +08:00
一维Tensor数据类型与 `x` 相同。
**异常:**
2022-05-25 14:16:31 +08:00
- **TypeError** - `x``mask` 不是Tensor。
- **TypeError** - `mask` 不是bool类型的Tensor。