Función Tensorflow.js tf.variableGrads()

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 .variableGrads() se usa para calcular y devolver el gradiente de f(x) en comparación con la lista indicada de variables manejables que presenta el parámetro varList . Además, si no se proporciona la lista, por defecto son todas las variables manejables.

Sintaxis:  

tf.variableGrads(f, varList?)

Parámetros:  

• f: Es la función indicada que se va a ejecutar. Donde, f() debe devolver un escalar. Es de tipo (() => tf.Scalar).
• varList: es la lista establecida de variables que se utilizan para calcular los gradientes en comparación y, de forma predeterminada, son todas las variables manejables. Es de tipo tf.Variable[].

Valor devuelto: Devuelve un valor que es de tipo tf.Scalar y también devuelve graduados que es de tipo {[name: string]: tf.Tensor}.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining list of variables
const p = tf.variable(tf.tensor1d([9, 6]));
const q = tf.variable(tf.tensor1d([7, 8]));
  
// Defining tf.tensor1d
const r = tf.tensor1d([3, 4]);
  
// Defining the function that is to
// be executed
const fn = () => p.add(r.square()).mul(q.add(r)).sum();
  
// Calling tf.variableGrads method
const {val, grads} = tf.variableGrads(fn);
  
// Printing output
Object.keys(grads).forEach(
    variable_Name => grads[variable_Name].print());

Producción:

Tensor
    [10, 12]
Tensor
    [18, 22]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining list of variables containing
// float values
const p = tf.variable(tf.tensor1d([3.1, 5.2]));
const q = tf.variable(tf.tensor1d([4.4, 6.7]));
  
// Defining tf.tensor1d with float values
const r = tf.tensor1d([7.1, 3.2]);
  
// Calling tf.variableGrads method
const {val, grads} = tf.variableGrads(
    () => p.add(r.square()).mul(q.add(r)).sum());
  
// Printing output
Object.keys(grads).forEach(
    variable_Name => grads[variable_Name].print());

Producción:

Tensor
    [11.5, 9.8999996]
Tensor
    [53.5099983, 15.4400005]

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