Función Tensorflow.js tf.unsortedSegmentSum()

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

La función .unsortedSegmentSum() se usa para calcular la suma a través de las partes de un tf.Tensor.

Sintaxis:  

tf.unsortedSegmentSum(x, segmentIds, numSegments)

Parámetros:  

• x: Es la entrada de tensor indicada que se va a sumar a través de sus partes y puede ser de tipo tf.Tensor, TypedArray o Array.
• segmentIds: Es el tf.Tensor1D indicado cuyo orden es equivalente al orden del tamaño de x a través del eje. Asigna cada elemento de x a una porción. Puede ser de tipo tf.Tensor1D, TypedArray o Array.
• numSegments: es el número indicado de segmentIds separados, y es de tipo number.

Valor devuelto: Devuelve el objeto tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input, segmentIds 
// and numSegments 
const y = tf.tensor1d([5, 6, 7, 8]);
const segmIds = tf.tensor1d([3, 1, 2, 0], 'int32');
const numSegm = 4;
  
// Calling tf.unsortedSegmentSum() method
// And printing output
y.unsortedSegmentSum(segmIds, numSegm).print();

Producción:

Tensor
    [8, 6, 7, 5]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.unsortedSegmentSum() method
var res = tf.unsortedSegmentSum(
    tf.tensor1d([1, 9]), 
    tf.tensor1d([1, 0], 'int32'), 3
);
  
// Printing output
res.print();

Producción:

Tensor
    [9, 1, 0]

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