目录 1、cond_fn()函数代码 2、softmax与log_softmax函数 1、cond_fn()函数代码 def cond_fn(x, t, y=None): assert y is not None with th.enable_grad(