Función Tensorflow.js tf.split()

. También ayuda a los desarrolladores a desarrollar modelos ML en lenguaje JavaScript y puede usar ML directamente en el navegador o en Node.js.

La función tf.split() se usa para dividir un tf.tensors en subtensores.

Sintaxis:

tf.split (x, numOrSizeSplits, axis?)

Parámetros:

• x: El tensor de entrada a dividir.
• numOrSizeSplits: puede ser un número que indica el número de divisiones o puede ser el arreglo en el que se proporcionan los tamaños para cada tensor de salida
• eje: Es un eje de la dimensión a lo largo del cual dividir.

Valor devuelto: Devuelve tf.Tensor[].

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor2d([10, 20, 50, 60,
                       30, 40, 70, 80], [2, 4]);
  
// Split the tensor
const [gfg, geeks] = tf.split(x, 2, 1);
  
gfg.print();
  
geeks.print();

Producción:

Tensor
    [[10, 20],
     [30, 40]]
Tensor
    [[50, 60],
     [70, 80]]

Ejemplo 2: En este ejemplo, hacer la división con el eje como segundo parámetro como una array.

const x = tf.tensor2d([10, 30, 50, 70, 20, 40, 60, 80], [2, 4]);
  
const [gfg, gfg1, geeks] = tf.split(x, [1, 2, 1], 1);
  
gfg.print();
  
gfg1.print();
  
geeks.print();

Producción:

Tensor
    [[10],
     [20]]
Tensor
    [[30, 50],
     [40, 60]]
Tensor
    [[70],
     [80]]

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