调用一次model.predict后获取层的张量值

时间:2020-03-03 13:04:36

标签: python keras-layer

运行以下代码后,如何获取神经网络中一层的输出激活的张量。

long sum = (long)intTemp + 5;

1 个答案:

答案 0 :(得分:0)

以下是使用Keras' functional API的简短示例:

inputs = Input(shape=(10,))
x = Dense(5)(inputs)
latent_outputs = Dense(2)(x)
x = Dense(5)(latent_outputs)
outputs = Dense(10)(x)

encoder = Model(inputs, latent_outputs)
autoencoder = Model(inputs, outputs)

autoencoder.compile(optimizer='adam', loss='mse')
autoencoder.fit(np.zeros((10, 10)), np.zeros((10, 10)))

latent_output = encoder.predict(np.zeros((1, 10)))
print(latent_output)