Как получить доступ к полной матрице z закодированного слоя в обучающем автоэнкодере, используя тензор потока?

Question

По сути, мне нужно получить значения кода z в виде полной матрицы.Например, у меня есть 60 000 изображений MNIST с размером 28 X 28 и размером слоя кода 32.

Итак, я ожидаю, что мой результат будет 60 000 X 32 вместо 60 000 X 784.

Я пытался:

  for layer in autoencoder.layers:
  print(layer.get_output_at(0)) 

И результат:

  Tensor("input_1:0", shape=(?, 784), 
  dtype=float32)
  Tensor("dense_1/Relu:0", shape=(?, 32), 
  dtype=float32)
  Tensor("dense_2/Sigmoid:0", shape=(?, 
  784), dtype=float32)

Но мне нужны значения на каждом слое.