Могу ли я выбрать данные из выходных данных модели кодировщика с tfp.layers.MultivariateNormalTriL в качестве выходных данных?

Question

У меня есть модель кодировщика-декодера, которая хорошо обучается. теперь я хочу сгенерировать выборки из вывода кодировщика, который является слоем tfp. возможно ли это?

original_inputs = tf.keras.Input(shape=(original_dim,), name='encoder_input')
x = tf.keras.layers.Dense(intermediate_dim, activation=activation)(original_inputs)
x = tf.keras.layers.Dense(intermediate_dim, activation=activation)(x)
x = tf.keras.layers.Dense(intermediate_dim, activation=activation)(x)
z = tf.keras.Sequential([
    tf.keras.layers.Dense(tfp.layers.MultivariateNormalTriL.params_size(latent_dim)),
    tfp.layers.MultivariateNormalTriL(latent_dim, activity_regularizer=tfp.layers.KLDivergenceRegularizer(prior)),
])(x)
encoder = tf.keras.Model(inputs=original_inputs, outputs=z, name='encoder')