def vae_view(head):
return ag_tuple((np.reshape(head[:latent_size], latent_shape),
np.reshape(head[latent_size:], obs_shape)))
def vae_view(head):
return ag_tuple(
(np.reshape(head[:z_size],
z_shape), np.reshape(head[z_size:], shape)))
def vae_view(head):
return ag_tuple(
(np.reshape(head[:latent1_size], latent1_shape),
np.reshape(head[latent1_size:latent1_size + latent2_size],
latent2_shape),
np.reshape(head[latent1_size + latent2_size:], (batch_size, ))))