Función Tensorflow.js tf.keep()

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 .keep() se usa para evitar que una entrada de tensor que se forma dentro de un método tf.tidy() se descarte espontáneamente.

Sintaxis:  

tf.keep(result)

Parámetros:  

• resultado: es la entrada del tensor indicada que se debe evitar que se descarte.

Valor devuelto: Devuelve el objeto tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Declaring a variable
let res1;
  
// Calling tidy method
const res2 = tf.tidy(() => {
     
  // Defining result parameter
  const result = tf.scalar(121);
     
  // Calling tf.keep() method
  res1 = tf.keep(result.sqrt());
});
  
// Printing output
res1.print();

Producción:

Tensor
    11

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Declaring a variable
let res1;
  
// Calling tidy method
const res2 = tf.tidy(() => {
     
  // Calling tf.keep() method with its
  // parameter
  res1 = tf.keep(tf.tensor1d(
    [1.3, 0.5, 0, NaN, null, -.5]).cos());
});
  
// Printing output
res1.print();

Producción:

Tensor
    [0.2675007, 0.8775977, 1, NaN, 1, 0.8775977]

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