Searched refs:batch_shape (Results 1 – 16 of 16) sorted by relevance
106 self._batch_shape = self.distribution.batch_shape145 if self.batch_shape is None or self.bijector.batch_shape is None:147 bijector_shape_tensor = self.fill_base(self.dtype, self.bijector.batch_shape, 0.0)148 dist_shape_tensor = self.fill_base(self.dtype, self.batch_shape, 0.0)
244 batch_shape = self.shape(concentration1 + concentration0)245 nan = self.fill(self.dtype, batch_shape, np.nan)344 batch_shape = self.shape(concentration1 + concentration0)345 origin_shape = shape + batch_shape
312 batch_shape = self.shape(mean + sd)313 origin_shape = shape + batch_shape
355 batch_shape = self.shape(concentration + rate)356 origin_shape = shape + batch_shape
353 batch_shape = self.shape(loc + scale)354 origin_shape = shape + batch_shape
347 batch_shape = self.shape(loc + scale)348 origin_shape = shape + batch_shape
396 batch_shape = self.shape(probs)[:-1]398 sample_shape = shape + batch_shape
135 def batch_shape(self): member in Distribution
87 batch_shape = input_shape[:-2]89 diag_shape = batch_shape + (_get_min(matrix_shape),)
64 str_info = 'batch_shape = {}'.format(self.batch_shape)
66 self._batch_shape = self.bijector.batch_shape
120 def batch_shape(self): member in Bijector146 self.parameter_type, self.batch_shape, 0.0)
108 str_info = 'batch_shape = {}'.format(self.batch_shape)
102 str_info = 'batch_shape = {}'.format(self.batch_shape)
137 str_info = 'batch_shape = {}'.format(self.batch_shape)