javascript - tensorflowjs 如何在 cnn 预测中获取内层输出

Question

我正在查看 tfjs 中的 tensorflow.js CNN 示例.

可在此处找到测试库:testing repo .

有什么方法可以从每一层获得输出吗？

 async showPredictions() {
    const testExamples = 1;
    // const testExamples = 100;
    const batch = this.data.nextTestBatch(testExamples);

    tf.tidy(() => {
        const output: any = this.model.predict(batch.xs.reshape([-1, 28, 28, 1]));

        output.print();
        const axis = 1;
        const labels = Array.from(batch.labels.argMax(axis).dataSync());
        const predictions = Array.from(output.argMax(axis).dataSync());

        // ui.showTestResults(batch, predictions, labels);
    });
}

上面是tfjs例子中的预测方法，但是只打印了最后一层。如何在预测中获得每一层(包括转换层、最大池化层和全连接层)的输出？

Answer 1

要获取所有内层，可以使用模型的属性layers。获得图层后，您可以使用每个图层的属性 input 和 output 来定义新模型或使用 apply方法。

有人问过类似的问题here和 there