You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.MaskedFill.rst 1.0 kB

4 years ago
123456789101112131415161718192021222324252627
  1. mindspore.ops.MaskedFill
  2. =========================
  3. .. py:class:: mindspore.ops.MaskedFill()
  4. 将掩码位置为True的位置填充指定的值。
  5. `input` 和 `mask` 的shape需相同或可广播。
  6. **输入:**
  7. - **input** (Tensor) - 输入Tensor,其数据类型为float16、float32、int8、或int32。
  8. - **mask** (Tensor[bool]) - 输入的掩码,其数据类型为bool。
  9. - **value** (Union[float, Tensor]) - 用来填充的值,只支持0维Tensor或float。
  10. **输出:**
  11. Tensor,输出与输入的数据类型和shape相同。
  12. **异常:**
  13. - **TypeError** - `input` 或 `mask` 不是Tensor。
  14. - **TypeError** - `value` 既不是float也不是Tensor。
  15. - **TypeError** - `input` 或 `value` 的数据类型不是float16、float32、int8、或int32。
  16. - **TypeError** - `value` 的数据类型与 `input` 不同。
  17. - **TypeError** - `mask` 的数据类型不是bool。
  18. - **ValueError** - `input` 和 `mask` 的shape不可广播。