INNER CODE UNIT · Python

backward

junshutang/Make-It-3D · activation.py:14

    def backward(ctx, g):
        x = ctx.saved_tensors[0]
        return g * torch.exp(x.clamp(max=15))

trunc_exp = _trunc_exp.apply

View source record →