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)