Función Tensorflow.js tf.grads()

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. 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.grads() toma una función f(x) y devuelve una función gx

Sintaxis :

tf.grads (f)

Parámetros :

• f: Esta es la función dada para la cual se calculan los gradientes.

Valor devuelto : Devuelve una array.

Ejemplo 1:

// Importing the @tensorflow/tjs library
const tf=require("@tensorflow/tfjs")
const f = (a, b) => b.add(a);
  
// Grad function is used
const g = tf.grads(f);
  
// Tensor is declared
const a = tf.tensor1d([5, 6]);
const b = tf.tensor1d([-10, -20]);
  
// Variables are defined
const [gfg1] = g([b, a]);
  
// Variable is printed
gfg1.print();

Producción:

Tensor
    [1, 1]

Ejemplo 2:

// Importing the @tensorflow/tfjs library
const tf=require("@tensorflow/tfjs")
const f = (a) => a.mul(8);
  
// Grad function is used
const g = tf.grads(f);
  
// Tensor is declared
const a = tf.tensor1d([50, 60]);
  
// Variables are defined
const [gfg1] = g([a]);
  
// Variable is printed
gfg1.print();

Producción:

Tensor
    [8, 8]

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