torch._assert

torch._assert(condition, message)

一个围绕 Python 的 assert 的封装，可以进行符号追踪。