Método Tensorflow.js tf.layers build()

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 .build() se usa para crear los pesos de la capa indicada. Este método debe aplicarse en todas las capas que soportan pesos. Además, se invoca cuando se invoca el método apply() para generar los pesos.

Sintaxis:

build(inputShape)

Parámetros:

• inputShape: es la forma indicada o una array de formas que no se modifica.

Valor devuelto: Devuelve nulo.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential();
  
// Adding a layer
model.add(tf.layers.dense({units: 1, inputShape: [3]}));
  
// Defining input
const input = tf.input({shape: [6, 2, 6]});
  
// Calling build method with its 
// parameter
model.layers[0].build([input.Shape]);
  
// Printing output
console.log(JSON.stringify(input.shape));
model.layers[0].getWeights()[0].print();

Producción:

[null,6,2,6]
Tensor
    [[-0.3726568],
     [0.7343086 ],
     [-0.2459907]]

Aquí, el método getWeights() se usa para imprimir los pesos.

Ejemplo 2:

// Importing the tensorflow.js library
//import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units: 1, inputShape: [2]}));
model.add(tf.layers.dense({units: 2}));
  
  
// Defining inputs
const input1 = tf.input({shape: [1, 2]});
const input2 = tf.input({shape: [1.7, 2.7, 6.5]});
  
// Calling build method with its 
// parameter
model.layers[0].build([input1.Shape]);
model.layers[1].build([input2.Shape]);
  
// Printing outputs
console.log(JSON.stringify(input1.shape));
console.log(JSON.stringify(input2.shape));
model.layers[0].getWeights()[0].print();
model.layers[1].getWeights()[0].print();

Producción:

[null,1,2]
[null,1.7,2.7,6.5]
Tensor
    [[0.6224715],
     [1.2144204]]
Tensor
     [[0.8342852, 0.4770206],]

Referencia: https://js.tensorflow.org/api/latest/#tf.layers.Layer.build