¿Cómo puedo obtener el resultado de cada capa en Tensorflow JS (Javascript)?

Por ahora estoy haciendo esto:

 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