Función Tensorflow.js tf.denseBincount()

Tensorflow.js es un conjunto de herramientas de código abierto desarrollado por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el navegador o en la plataforma del Node. También permite a los desarrolladores crear modelos de aprendizaje automático en JavaScript y utilizarlos directamente en el navegador o con Node.js.

La función tf.denseBincount() se usa para construir un tensor del tamaño y tipo de datos especificados. La suma de números en un índice del tensor de peso dado que corresponde al índice del número índice en el tensor de entrada serían los valores del tensor.

Sintaxis:

tf.denseBincount (x, weights, size, binaryOutput?) 

Parámetros:

• x: Este es el tensor de entrada. Puede ser un tensor 1D o 2D.
• pesos: Es el tensor de pesos que debe tener la misma forma que x, o debe ser un tensor de longitud 0, en cuyo caso actúa como si todos los pesos fueran iguales a 1.
• tamaño: El tamaño del tensor de salida.
• Salida binaria: Esto es opcional. Se utiliza para especificar si el kernel debe contar el número de ocurrencias o la aparición. Falso es el valor predeterminado.

Devoluciones: tf.Tensor1D o tf.Tensor2D

Ejemplo 1:

import * as tf from "@tensorflow/tfjs";
  
const x = tf.tensor([7, 10, 3, 6, 8, 6, 4, 
    10, 5, 9, 2, 5, 9, 6, 1, 4, 10, 5,], 
    [1, 18], 'int32');
      
const weight = [];
const size = 11;
const out = tf.denseBincount(x, weight, size);
console.log(out.print());

Producción:

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

Ejemplo 2:

import * as tf from "@tensorflow/tfjs";
  
const x = tf.tensor([1, 2, 9, 6, 5, 4, 
    7, 4, 7, 4, 3], [1, 11], 'int32');
const weight = [0, 2, 5, 8, 9, 3, 5, 5, 3, 8, 2];
const size = 10;
const out = tf.denseBincount(x, weight, size);
console.log(out.print());

Producción:

Tensor
   [[0, 0, 2, 2, 16, 9, 8, 8, 0, 5],]

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