Función Tensorflow.js tf.stack()

. Ayuda a los desarrolladores a desarrollar modelos ML en JavaScript y usar ML directamente en el navegador o en Node.js.

La función tf.stack() se usa para crear una pila de tf,tensores en un tf.tensor de rango r+1.

Sintaxis:

tf.stack(tensors, axis)

Parámetros: esta función acepta dos parámetros, como se mencionó anteriormente y se analiza a continuación.

• tensores:
• eje: Es un eje longitudinal de la pila.

Valor devuelto: Devuelve tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Making a tensor a
const a = tf.tensor1d([99, 999, 999]);
  
// Making a tensor b
const b = tf.tensor1d([322, 411, 888]);
  
// Making a tensor c
const c = tf.tensor1d([523, 622, 666]);
  
// Printing the stack
tf.stack([a, b, c]).print();

Producción:

Tensor
    [[99 , 999, 999],
     [322, 411, 888],
     [523, 622, 666]]

Ejemplo 2: En este ejemplo, hacer la pila con el eje como segundo parámetro.

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Making a tensor a
const a = tf.tensor1d([99, 999, 999]);
  
// Making a tensor b
const b = tf.tensor1d([322, 411, 888]);
  
// Making a tensor c
const c = tf.tensor1d([523, 622, 666]);
  
// Printing the stack
tf.stack([a, b, c], 1).print();

Producción:

Tensor
    [[99 , 322, 523],
     [999, 411, 622],
     [999, 888, 666]]

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