def __init__(self,
temperature,
logits,
group_ndims=0,
is_reparameterized=True,
use_path_derivative=False,
check_numerics=False,
**kwargs):
self._logits = tf.convert_to_tensor(logits)
self._temperature = tf.convert_to_tensor(temperature)
param_dtype = assert_same_float_dtype([
(self._logits, 'Concrete.logits'),
(self._temperature, 'Concrete.temperature')
])
self._logits, self._n_categories = assert_rank_at_least_one(
self._logits, 'Concrete.logits')
self._temperature = assert_scalar(self._temperature,
'Concrete.temperature')
self._check_numerics = check_numerics
super(Concrete, self).__init__(dtype=param_dtype,
param_dtype=param_dtype,
is_continuous=True,
is_reparameterized=is_reparameterized,
use_path_derivative=use_path_derivative,
group_ndims=group_ndims,
**kwargs)
def __init__(self,
temperature,
logits,
group_ndims=0,
is_reparameterized=True,
check_numerics=False,
**kwargs):
self._logits = tf.convert_to_tensor(logits)
self._temperature = tf.convert_to_tensor(temperature)
param_dtype = assert_same_float_dtype([
(self._logits, 'BinConcrete.logits'),
(self._temperature, 'BinConcrete.temperature')
])
self._temperature = assert_scalar(self._temperature,
'BinConcrete.temperature')
self._check_numerics = check_numerics
super(BinConcrete,
self).__init__(dtype=param_dtype,
param_dtype=param_dtype,
is_continuous=True,
is_reparameterized=is_reparameterized,
group_ndims=group_ndims,
**kwargs)