tensorflowJS cant figure out arrays

I am trying to input an array of data and output a prediction or a guess for what the result should be for the given data, I do not know much about neural nets but have a basic understanding on how their training works... I was hoping that it would see the two given inputs combined add up to the output in the three examples I provided for it such as [1, 2] should = 3 but I have hit a wall with inputting arrays and am getting the fallowing error

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

any ideas someone can give me on how I should be going about this would be etremely helpful -.-

import tf from '@tensorflow/tfjs';

// create model object
const model = tf.sequential();
    units: 1,
    inputShape: [1]

// compile model object
    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([


async function firstFunction(_callback) {

    // Train model with fit().method
    await model.fit(xs, ys, {
        epochs: 100000

function secondFunction() {

    //We wait for the call back from first function.
    firstFunction(function() {

        // We run the prediction.                    
        prediction = model.predict(tf.tensor2d([
            [6, 9],


How many English words
do you know?
Test your English vocabulary size, and measure
how many words do you know
Online Test
Powered by Examplum