How can I get the output of each layer in Tensorflow JS (Javascript)?
For now I am doing this:
const input_nodes = 5;
const hidden_nodes = 8;
const output_nodes = 4;
let inputs = [1, 2, 3, 4, 5];
const xs = tf.tensor2d([inputs]);
let model1 = tf.sequential();
model1.add(tf.layers.dense({ units: hidden_nodes, inputShape: [input_nodes], activation: 'sigmoid' }));
console.log(model1.predict(xs).dataSync()); // Output for Layer 1
let model2 = tf.sequential();
model2.add(tf.layers.dense({ units: hidden_nodes, inputShape: [input_nodes], activation: 'sigmoid' }));
model2.add(tf.layers.dense({ units: output_nodes, activation: 'softmax' }));
console.log(model2.predict(xs).dataSync()); // Output for Layer 1