def data_initializer_prior(data_segments, data_labels):
# Input data
segments_initializer = tf.placeholder_with_default(
tf.zeros(data_segments.shape, tf.int32),
shape=data_segments.shape,
name='segments_initializer')
labels_initializer = tf.placeholder_with_default(
tf.zeros(data_labels.shape, tf.int32),
shape=data_labels.shape,
name='labels_initializer')
input_segments = tf.Variable(
segments_initializer, trainable=False,
collections=[tf.GraphKeys.LOCAL_VARIABLES], name='input_segments')
input_labels = tf.Variable(
labels_initializer, trainable=False,
collections=[tf.GraphKeys.LOCAL_VARIABLES], name='input_labels')
return (segments_initializer, labels_initializer, input_segments, input_labels)
评论列表
文章目录