Python中 assert断言声明，遇到错误则立即返回

80 阅读 0 评论 53 点赞

我是靠谱客的博主勤奋烧鹅，最近开发中收集的这篇文章主要介绍Python中 assert断言声明，遇到错误则立即返回，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

在使用python语言开发深度学习模型时，经常会遇到模型运行结束时才会发现的崩溃状态，或者得到的结果不是预想的那样。这就需要在模型中善于使用assert语句，来判断某些变量的类型或者shape等是否满足要求。

用法为：

assert expression(你想判断的内容)

# 等价于
if not expression:
    raise AssertionError

例如：在逻辑回归中，z = W的转置*X，样本X含有五个特征，则我们在设置W时。要一直保证其shape为[5,1]


>>> import numpy as np
>>> w = np.zeros((5,1))
>>> w
array([[0.],
       [0.],
       [0.],
       [0.],
       [0.]])
>>> assert(w.shape == (5,1))
>>> assert(w.shape == (2,1))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError