def _compose_save_list_secure(self, *pairs):
with tf.name_scope('secure_save_list'):
save_list = list()
for pair in pairs:
variables = flatten(pair[0])
new_values = flatten(pair[1])
s_list = self._recurrent_assign_loop_with_dependencies(
variables, new_values, [])
save_list.extend(s_list)
return save_list
def _compose_save_list(self,
*pairs):
#print('start')
with tf.name_scope('save_list'):
save_list = list()
for pair in pairs:
#print('pair:', pair)
variables = flatten(pair[0])
#print(variables)
new_values = flatten(pair[1])
for variable, value in zip(variables, new_values):
name = self._extract_op_name(variable.name)
save_list.append(tf.assign(variable, value, name='assign_save_%s' % name))
return save_list
def _compose_reset_list(self, *args):
with tf.name_scope('reset_list'):
reset_list = list()
flattened = flatten(args)
for variable in flattened:
shape = variable.get_shape().as_list()
name = self._extract_op_name(variable.name)
reset_list.append(tf.assign(variable, tf.zeros(shape), name='assign_reset_%s' % name))
return reset_list
def _compose_randomize_list(self, *args):
with tf.name_scope('randomize_list'):
randomize_list = list()
flattened = flatten(args)
for variable in flattened:
shape = variable.get_shape().as_list()
name = self._extract_op_name(variable.name)
assign_tensor = tf.truncated_normal(shape, stddev=1.)
randomize_list.append(tf.assign(variable, assign_tensor, name='assign_reset_%s' % name))
return randomize_list
def _compose_reset_list(self, saved_last_sample_predictions, *args):
with tf.name_scope('reset_list'):
reset_list = list()
name = self._extract_op_name(saved_last_sample_predictions.name)
reset_list.append(
tf.assign(saved_last_sample_predictions,
np.tile(
char_2_base_vec(
self._character_positions_in_vocabulary,
self._replica_delimiter), (1, 1)),
name='assign_reset_%s' % name))
flattened = flatten(args)
for variable in flattened:
shape = variable.get_shape().as_list()
name = self._extract_op_name(variable.name)
reset_list.append(
tf.assign(variable,
tf.zeros(shape),
name='assign_reset_%s' % name))
return reset_list
