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

17 lines
661 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-07-26 10:27:56 +08:00
输入:
- **x** (Tensor) - 它的shape是 :math:`(x_1, x_2, ..., x_R)`
- **mask** (Tensor[bool]) - 它的shape是 :math:`(x_1, x_2, ..., x_R)`
2022-07-26 10:27:56 +08:00
输出:
一维Tensor数据类型与 `x` 相同。
2022-07-26 10:27:56 +08:00
异常:
- **TypeError** - `x``mask` 不是Tensor。
- **TypeError** - `mask` 不是bool类型的Tensor。