Tensorflow JS: разные формы для вывода

Question

Я совершенно новичок в Tensorflow.

Моя цель довольно проста: у меня есть трехмерный тензор в качестве входного / обучающего значения, и я хотел бы "отобразить" его на 1d выходной тензор .

При запуске моей модели я получаю сообщение об ошибке, что 1-й выходной тензор не может быть назначен определенной [5, 5] форме:

const model = tf.sequential({
  layers: [
    tf.layers.dense({
      inputShape: [5, 5],
      units: 32,
      activation: "relu"
    }),
    tf.layers.dense({ units: 1, activation: "softmax" }),
  ]
});

Возможно ли иметь разные формы для вывода / ввода? Я хочу, чтобы 3d-тензор был похож на «группы чисел», разрешенные в одно (1d-тензорное) число.

Answer 1

Чтобы отобразить многомерный тензор (превосходящий 1) на 1d-тензор, необходимо использовать сплющенные слои между

const model = tf.sequential({
  layers: [
    tf.layers.dense({
      inputShape: [5, 5],
      units: 32,
      activation: "relu"
    }),
    tf.layers.flatten(),
    tf.layers.dense({ units: 1, activation: "softmax" }),
  ]
});