. 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:
Javascript
// 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.
Javascript
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