torch._assert

torch._assert(condition, message)

Python 的 assert 的包装器，具有符号可追踪性。