Función Tensorflow.js tf.concat()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

La función tf.concat() se usa para concatenar la lista de tensores especificados a lo largo del eje dado.

Sintaxis:

tf.concat (tensors, axis)

Parámetros: Esta función acepta dos parámetros que se ilustran a continuación:

• tensores: Es una lista de tensores especificados para concatenar.
• eje: Es el eje a lo largo del cual se está realizando el proceso de concatenación. Es un parámetro opcional y su valor por defecto es 0.

Valor de Retorno: Devuelve un Tensor de tensores concatenados. 

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing two tensors to concatenate
const A = tf.tensor1d([0, 2, 4]);
const B = tf.tensor1d([1, 3, 5]);
  
// Calling the .concat() function over
// the above tensors as its parameters
A.concat(B).print();

Producción:

Tensor
   [0, 2, 4, 1, 3, 5]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing three 2-D tensors to concatenate
const A = tf.tensor2d([[0, 2], [1, 3]]);
const B = tf.tensor2d([[4, 6], [5, 7]]);
const C = tf.tensor2d([[8, 10], [9, 11]]);
  
// Initializing axis parameter
const axis = 1;
  
// Calling the .concat() function over
// the above tensors and axis as its parameters
tf.concat([A, B, C], axis).print();

Producción:

Tensor
   [[0, 2, 4, 6, 8, 10],
    [1, 3, 5, 7, 9, 11]]

Referencia: https://js.tensorflow.org/api/latest/#concat