Function modulate

Large-DiT-ImageNet/models.py:28–29

(x, shift, scale)

def modulate(x, shift, scale):
    return x * (1 + scale.unsqueeze(1)) + shift.unsqueeze(1)


#############################################################################