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

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 .countParams() se usa para encontrar el recuento absoluto de números como float32 , int32 en los pesos indicados.

Sintaxis:

countParams()

Parámetros: este método no contiene ningún parámetro.

Valor devuelto: Devuelve número.

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: 2, inputShape: [11]}));
  
// Calling setWeights() method
model.layers[0].setWeights(
    [tf.truncatedNormal([11, 2]), tf.zeros([2])]);
  
// Calling countParams() method and also
// Printing output
console.log(model.layers[0].countParams());

Salida: aquí, el método truncatedNormal() se usa para crear un tf.Tensor junto con valores que se muestrean de una distribución normal truncada, el método zeros() se usa para crear un tf.Tensor junto con todos los elementos que se establecen en 0 y el método setWeights() se utiliza para establecer los pesos.

24

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: [5], batchSize: 1, dtype: 'int32'}));
model.add(tf.layers.dense({units: 2, inputShape: [6], batchSize: 5}));
model.add(tf.layers.dense({units: 3, inputShape: [7], batchSize: 8}));
model.add(tf.layers.dense({units: 4, inputShape: [8], batchSize: 12}));
  
// Calling setWeights() method
model.layers[0].setWeights([tf.ones([5, 1]), tf.zeros([1])]);
model.layers[1].setWeights([tf.ones([1, 2]), tf.zeros([2])]);
  
// Calling countParams() method and also
// Printing outputs
console.log(model.layers[0].countParams());
console.log(model.layers[1].countParams());
console.log(model.layers[2].countParams());

Salida: aquí, el método ones() se usa para crear un tf.Tensor junto con todos los elementos que se establecen en 1.

6
4
9

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