Estoy tratando de ingresar una matriz de datos y generar una predicción o una suposición de cuál debería ser el resultado para los datos dados, no sé mucho sobre redes neuronales, pero tengo una comprensión básica de cómo funciona su entrenamiento ... Estaba con la esperanza de que las dos entradas proporcionadas combinadas se sumen a la salida en los tres ejemplos que proporcioné, como [1, 2] debería = 3, pero me he topado con una pared al ingresar matrices y obtengo el siguiente error.

 UnhandledPromiseRejectionWarning: Error: input expected a batch of elements where each example has shape [1] (ie,tensor shape [*,1]) but the input received an input with 3 examples, each with shape [2] (tensor shape [3,2])

cualquier idea que alguien pueda darme sobre cómo debería hacerlo sería de gran ayuda -.-

 import tf from '@tensorflow/tfjs'; // create model object const model = tf.sequential(); model.add(tf.layers.dense({ units: 1, inputShape: [1] })); // compile model object model.compile({ optimizer: 'sgd', loss: 'meanSquaredError' }); // training datasets // input data for training const xs = tf.tensor2d([ [1, 2], [5, 6], [3, 4] ]); // output data for training const ys = tf.tensor1d([ 3, 11, 7 ]); async function firstFunction(_callback) { // Train model with fit().method await model.fit(xs, ys, { epochs: 100000 }) _callback(); } function secondFunction() { //We wait for the call back from first function. firstFunction(function() { // We run the prediction. prediction = model.predict(tf.tensor2d([ [6, 9], ])).dataSync()[0] }); } secondFunction()