mindspore.dataset.transforms.Mask ================================= .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/dataset_transforms/mindspore.dataset.transforms.Mask.rst :alt: 查看源文件 .. py:class:: mindspore.dataset.transforms.Mask(operator, constant, dtype=mstype.bool_) 用给条件判断输入Tensor的内容,并返回一个掩码Tensor。Tensor中任何符合条件的元素都将被标记为True,否则为False。 参数: - **operator** (:class:`~.transforms.Relational`) - 关系操作符,可以取值为 ``Relational.EQ`` 、 ``Relational.NE`` 、 ``Relational.LT`` 、 ``Relational.GT`` 、 ``Relational.LE`` 、 ``Relational.GE`` 。以 ``Relational.EQ`` 为例,将找出Tensor中与 `constant` 相等的元素。 - **constant** (Union[str, int, float, bool]) - 与输入Tensor进行比较的基准值。 - **dtype** (:class:`mindspore.dtype`, 可选) - 生成的掩码Tensor的数据类型。默认值: ``mstype.bool_`` 。 异常: - **TypeError** - 参数 `operator` 类型不为 :class:`mindspore.dataset.transforms.Relational` 。 - **TypeError** - 参数 `constant` 类型不为str、int、float或bool。 - **TypeError** - 参数 `dtype` 类型不为 :class:`mindspore.dtype` 。