如何在 tensorflow.js 中为简单网络配置层？

Question

TLDR：任何人都可以提供调用和参数来重现下面 tensorflow.js 中的示例网络配置吗？提前致谢！

我正在尝试将我现有的神经网络转换为 tensorflow.js，但我发现 API 令人困惑并且文档并没有真正帮助（我是该领域的新手）。具体来说，当我添加图层时，我不确定应该发送什么值。

另外，我不太理解输入'shape'。什么形状？我更多的理解是长度，即输入节点的数量。

我正在使用顺序模型并调用 tf.layers.dense。我的输入是简单的浮点数组； XY 和 XYZ 坐标。关于 'units'，我不清楚 'inputShape' 与 'batchInput'。此外，当我显示 modelSummary 时，它列出了我不期望的“输出形状”和“# Of Params”的值。

Answer 1

原来没有离散的创建输入层步骤。相反，'input layer' 是网络输入的形状，设置为第一个添加层的属性。

所以在我的例子中，我向网络 A 添加一个密集层，向网络 B 添加两个密集层，如下所示：

// Network A
modelA.add(tf.layers.dense({
  inputShape: 2,
  units: 1,
  useBias: true
}));

// Network B
modelB.add(tf.layers.dense({
  inputShape: 3,
  units: 4
}));

modelB.add(tf.layers.dense({
  units: 2,
  useBias: true
}));

如何在 tensorflow.js 中为简单网络配置层？

How to configure layers for simple networks in tensorflow.js?

tensorflow.js