print all layers output

后端未结

关注

 2  854

清酒与你

Given the following model, how to print all layers values ?

相关标签:

2条回答

庸人自扰

2021-01-26 05:35
You could so something like
```
    for(var i = 0; i < tf.layers.length; i++)
      model.predict(tf.layers[i].value).print();
      // OR
      model.predict(tf.layers[i].inputs).print();
```
I don't know how is the structure of your array, but something like that could work.
0 讨论(0)
发布评论:

提交评论
- 加载中...

北海茫月

2021-01-26 05:35

To print layers, one needs to define the layers to output in the model configuration, using, outputs property. Using destructuring assignement on model.predict() one could retrieve the intermediate layers to output

const input = tf.input({shape: [5]});
        const denseLayer1 = tf.layers.dense({units: 10, activation: 'relu'});
        const denseLayer2 = tf.layers.dense({units: 2, activation: 'softmax'});
        const output1 = denseLayer1.apply(input);
        const output2 = denseLayer2.apply(output1);
        const model = tf.model({inputs: input, outputs: [output1, output2]});
        const [firstLayer, secondLayer] = model.predict(tf.ones([2, 5]));
        firstLayer.print();
        secondLayer.print()

<html>
  <head>
    <!-- Load TensorFlow.js -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.12.0"> </script>
  </head>

  <body>
  </body>
</html>

0 讨论(0)

热议问题