我试图了解 TensorFlow 的输出层有什么可能。具体来说,输出总是一个平面数组吗?
由于神经元(或 TF 中的“单元”)只有一个数字,并且只有一组输出,因此输出似乎必须具有单一维度。使用 one-hot 概率,这很容易理解。但是图像呢?
如果我的输出是一张图片,我可以让 TF 输出一个多维像素数组,例如 [[r0, g0, b0], [r1, g1, b1], ...]?如果是这样,该网络将如何构建?我将如何定义输出层的维度/形状?
我所知道的定义输出形状的唯一参数是这个,来自 tf.layers.dense,它看起来本质上是一维的:
- units (number) 正整数,输出空间的维度。
非常感谢您提供的任何帮助!
参考:https://js.tensorflow.org/api/latest/#layers.dense