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.valueAndGrad() se usa para devolver el gradiente de la función especificada f(x) con respecto a x junto con el valor de f().
Sintaxis:
tf.valueAndGrad (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 junto con el valor de f().
Ejemplo 1:
Javascript
// 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 .valueAndGrad() function over // the above function as its parameter which // calculates f'(x) = 2x const g = tf.valueAndGrad(f); // Initializing a Tensor of values at which // value of gradient is calculated const x = tf.tensor1d([0, 1, 2, 3]); // Returning the value of f() and // gradient of f(x) const {value, grad} = g(x); // Getting the value of f() console.log('value'); value.print(); // Getting the gradient of f(x) at // the above tensor values console.log('grad'); grad.print();
Producción:
value Tensor [0, 1, 4, 9] grad Tensor [0, 2, 4, 6]
Ejemplo 2:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Using a function f(x) = x^3 as the parameter // for the .valueAndGrad() function which // calculates f'(x) = 3x^2 const g = tf.valueAndGrad(x => x.pow(tf.scalar(3, 'int32'))); // Using a Tensor of values at which // value of gradient is calculated const {value, grad} = g(tf.tensor1d([-1, 0, 0.3, 4])); // Getting the value of f() console.log('value'); value.print(); // Getting the gradient of f(x) console.log('grad'); grad.print();
Producción:
value Tensor [-1, 0, 0.027, 64] grad Tensor [3, 0, 0.27, 48]
Referencia: https://js.tensorflow.org/api/latest/#valueAndGrad
Publicación traducida automáticamente
Artículo escrito por Kanchan_Ray y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA