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

【问题标题】：How to configure layers for simple networks in tensorflow.js?如何在 tensorflow.js 中为简单网络配置层？
【发布时间】：2020-12-12 05:05:15
【问题描述】：

TLDR：谁能提供将在 tensorflow.js 中重现以下示例网络配置的调用和参数？提前致谢！

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

另外，我不太了解输入的“形状”。什么形状？我把它理解为一个长度，即输入节点的数量。

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

【问题讨论】：

标签： tensorflow.js

【解决方案1】：

事实证明，没有离散的创建输入层步骤。相反，“输入层”是网络输入的形状，设置为第一个添加层的属性。

所以在我的例子中，我向网络 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
}));

【讨论】：