Función Tensorflow.js tf.grad()

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.grad() se usa para devolver el gradiente de la función especificada f(x) con respecto a x.

Sintaxis:  

tf.grad (f)

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

• f: la función especificada f(x) para la que se calcula el gradiente.

Valor devuelto: Devuelve el gradiente de la función especificada f(x) con respecto a x.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a function f(x) = x^2
const f = x => x.square();
  
// Calling the .grad() function which 
// calculates f'(x) and gives value as 2x
const g = tf.grad(f);
  
// Initializing a tensor of values for
// which gradient will be returned
const x = tf.tensor1d([1, 2, 3]);
  
// Getting the values of gradient of the
// function f(x) for the above specified
// tensor 
g(x).print();

Producción:

Tensor
   [2, 4, 6]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using the function f(x) = x^3 as the 
// parameter for the .grad() function which 
// calculates f'(x) and gives value as 3x^2
const g = tf.grad(x => x.pow(tf.scalar(3, 'int32')));
  
// Initializing a tensor of values for
// which gradient will be returned
const x = tf.tensor1d([0, 1, 2]);
  
// Getting the values of gradient of the
// function f(x) for the above specified
// tensor 
g(x).print();

Producción:

Tensor
   [0, 3, 12]

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