Función Tensorflow.js tf.eye()

Tensorflow.js es una biblioteca de código abierto para crear modelos de aprendizaje automático en Javascript que permite a los usuarios ejecutar los modelos directamente en el navegador.

El tf.eye() es una función definida en la clase tf . Tensor. Se utiliza para crear una array de identidad de filas y columnas específicas.

Una array identidad de forma [m, n] consta de valor uno en todos los elementos diagonales y cero en los lugares restantes.

Sintaxis:

tf.eye(numRows, numColumns, batchShape, dtype)

Parámetros:

• numRows: número de filas de la array de identidad que regresa.
• numColumns : número de columnas de la array de identidad de retorno. Es un parámetro opcional. Si no se especificó, entonces la forma de la array de identidad de retorno es [númRows, numRows].
• batchShape: es una array que define la forma que se agregará al comienzo de la forma de la array de identidad devuelta al replicar la array de identidad. Es un parámetro opcional. SupposebatchShape = [a, b], entonces la forma de la array de identidad que regresa es [a, b, numRows, numColumns]
• dtype: especificado como el tipo de datos de los elementos en la array de identidad de retorno. Es opcional.

Valor devuelto: Devuelve un tensor de array identidad.

Ejemplo 1: Crear una array de identidad de forma cuadrada

• Cree una array de identidad de forma [3, 3] definiendonumRows = 3 en el método tf.eye() .

// Dynamic loading the "@tensorflow/tfjs" module
const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
 
// Creating an identity matrix  of shape [3,3]
var matrix = tf.eye(numRows = 3)
   
// Printing the identity matrix
matrix.print()

Producción :

Tensor
    [[1, 0, 0],
     [0, 1, 0],
     [0, 0, 1]]

Ejemplo 2: Crear una array identidad de forma rectangular

• Cree una array de identidad de forma[3, 4] definiendo numRows = 3andnumColumns = 4 en el método tf.eye() .

// Dynamic loading the "@tensorflow/tfjs" module
const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
 
// Creating an identity matrix  of shape [3,4]
var matrix = tf.eye(numRows = 3, numColumns = 4)
   
// Printing the identity matrix
matrix.print()

Producción :

Tensor
    [[1, 0, 0, 0],
     [0, 1, 0, 0],
     [0, 0, 1, 0]]

Ejemplo 3: crear una array de identidad especificando batchShape

• Cree una array de identidad de forma [2, 3] definiendo numRows = 2 y numColumns = 3 en el método tf.eye() .
• Replique la array de identidad anterior 3 veces incluyendo batchShape = [3]
• Entonces la forma de la array de identidad que regresa es [3, 2, 3]

// Dynamic loading the "@tensorflow/tfjs" module
const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
 
// Creating an identity matrix  of shape [2,3]
// And include batchShape = [3] to replicate identity matrix 3 times
// Shape of returning tensor is [3, 2, 3]
var matrix = tf.eye(numRows = 2, numColumns = 3, batchShape = [3])
   
// Printing the identity matrix
matrix.print()

Producción :

Tensor
    [[[1, 0, 0],
      [0, 1, 0]],

     [[1, 0, 0],
      [0, 1, 0]],

     [[1, 0, 0],
      [0, 1, 0]]]

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