mindspore.ops.Assert

class mindspore.ops.Assert(summarize=3)[源代码]

判断给定条件是否为True，若不为True则以list的形式打印 input_data 中的Tensor，否则继续往下运行代码。

参数：

summarize (int, 可选) - 当判断结果为False时，打印 input_data 中每个Tensor的条目的数量。默认值：3。

输入：

condition ([Union[Tensor[bool], bool]]) - 需要进行判断的条件。
input_data (Union(tuple[Tensor], list[Tensor])) - 当 condition 被判断为False的时候将被打印的Tensor。

异常：

TypeError - summarize 的数据类型不是int。
TypeError - condition 的数据格式不是Tensor或bool。
TypeError - input_data 的数据格式不是list或tuple。

支持平台：

GPU CPU

样例：

>>> a = Tensor(np.array([-1, 0, 1, 2, 3]).astype(np.int32))
>>> b = Tensor(np.array([1, 2, 3, 4, 5]).astype(np.float32))
>>> assert1 = ops.Assert(3)
>>> assert1(False, [a, b])
For 'Assert' condition is false.
input data: [-1 0 1]
input data: [1 2 3]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "mindspore/ops/primitive.py", line 294, in __call__
    return _run_op(self, self.name, args)
  File "mindspore/common/api.py", line 99, in wrapper
    results = fn(*arg, **kwargs)
  File "mindspore/ops/primitive.py", line 743, in _run_op
    output = real_run_op(obj, op_name, args)
RuntimeError: assert failed