torch._assert

torch._assert(condition, message)

Python 中 assert 的包装器，可以进行符号跟踪。