def initialize(self, device, **kwargs):
super(StochasticPoolingBase, self).initialize(device=device, **kwargs)
if self.uniform is None:
self.uniform = Uniform(self)
if self.uniform.output_bytes < (self.output_size << 1):
if self.uniform.is_initialized:
raise ValueError(
"uniform is already initialized and does not have enough "
"output size")
self.uniform.output_bytes = self.output_size << 1
self.uniform.initialize(self.device)
