torch._assert

torch._assert(condition, message)

一个对 Python assert 的包装，它可以在符号层面进行追踪。