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:
Javascript
// 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:
Javascript
// 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
Publicación traducida automáticamente
Artículo escrito por nidhi1352singh y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA