def __init__(self, loc, scale, validate_args=None):
base_dist = Uniform(t.zeros(loc.shape),
t.ones(loc.shape),
validate_args=validate_args)
if not base_dist.batch_shape:
base_dist = base_dist.expand([1])
super(Logistic, self).__init__(
base_dist, [SigmoidTransform().inv,
AffineTransform(loc, scale)])
def expand(self, batch_shape):
return Uniform.expand(self, batch_shape, _instance=self)
