(None, 3, 2) just the example of input shape, you could use input shape (None, 256, 2), and output latent with(1, 64), 256 * 2 -> 64 is the dimension reduce!