Función Tensorflow.js tf.whereAsync()

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

La función tf.whereAsync() se usa para devolver el tensor 2-D de coordenadas de elementos verdaderos para la condición especificada. En el tensor devuelto, la primera dimensión, es decir, filas, especifica el número de elementos verdaderos, y la segunda dimensión, es decir, columnas, especifica las coordenadas de los elementos verdaderos, es decir, el tensor de salida tiene la forma de [numTrueElems, condition.rank].

Nota: La forma del tensor devuelto depende de los valores verdaderos presentes en la entrada. 

Sintaxis:  

tf.whereAsync (condition)

Parámetros: Esta función acepta un parámetro que se ilustra a continuación:

• condición: la condición de entrada es una array de valores booleanos.

Valor devuelto: Devuelve el tensor 2-D de coordenadas de elementos verdaderos para la condición especificada. 

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor for conditions
const cond = tf.tensor1d([false, true, true, false, true], 'bool');
  
// Calling the .whereAsync() function
const result = await tf.whereAsync(cond);
  
// Getting the 2-D tensor of true elements
result.print();

Producción:

Tensor
   [[1],
    [2],
    [4]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using a tensor of conditions as the
// parameter of the .whereAsync() function
const result = await tf.whereAsync(tf.tensor1d(
  [true, false, true, true, false], 'bool'));
  
// Getting the 2-D tensor of true elements
result.print();

Producción:

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

Referencia: https://js.tensorflow.org/api/1.0.0/#whereAsync