Función Tensorflow.js tf.initializers.identity()

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 clase Initializer es la clase base de todos los inicializadores en Tensorflow.js. Los inicializadores se utilizan para inicializar los tensores con los valores específicos. Devuelve el objeto tensor inicializado como lo especifica el inicializador. Entonces, en este artículo, vamos a ver cómo funciona el inicializador de identidad. Este es el inicializador que inicializó un nuevo objeto tensor con una array identidad. Solo se usa para arrays 2D.

Sintaxis:

tf.initializers.identity(Gain)

Parámetro:

• Ganancia : Es el factor de multiplicación que se aplica a la array identidad.

Valor devuelto: Devuelve tf.initializers.Initializer

Ejemplo 1: En este ejemplo, vamos a comprobar el uso independiente de la función identidad().

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Generates the identity matrix
const value=tf.initializers.identity(1.0)
 
// Print gain
console.log(value)

Producción :

{
 "gain": 1
}

Ejemplo 2: En este ejemplo, vamos a usar una array identidad con una capa densa usando la función identity() y dense().

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Define the input
const inp = tf.input({shape:[4]});
 
// Create identity matrix with gain 1
const value=tf.initializers.identity(1.0)
 
 
// Dense layer 1
const denseLayer1 = tf.layers.dense({
    units: 6,
    activation: 'relu',
    kernelInitialize: value
});
 
// Dense layer 2
const denseLayer2 = tf.layers.dense({
    units: 8,
    activation: 'softmax'
});
 
const out = denseLayer2.apply(denseLayer1.apply(inp));
 
//  Model creation
const model = tf.model({inputs:inp,outputs:out});
 
// Make prediction
console.log("Lets Make Some Prediction :")
model.predict(tf.ones([2, 4])).print();

Producción :

Lets Make Some Prediction :
Tensor
   [[0.1651815, 0.1695402, 0.0670628, 0.0771763, 
     0.1045933, 0.1027268, 0.1647871, 0.148932],
    [0.1651815, 0.1695402, 0.0670628, 0.0771763, 
     0.1045933, 0.1027268, 0.1647871, 0.148932]]

Referencia:   https://js.tensorflow.org/api/3.6.0/#initializers.identity