Tensorflow.js tf.LayersModel class .getLayer() Method

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"

// Defining model
const model = tf.sequential();

// Adding a layer
model.add(tf.layers.dense({units: 4, inputShape: [1]}));

// Calling getLayer() method
const layer_0 = model.getLayer(null, 0);

// Printing weights of the layer_0
// using getWeights() method
layer_0.getWeights()[0].print();

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"

// Defining model
const model = tf.sequential();

// Adding layers
model.add(tf.layers.dense({units: 4, inputShape: [1]}));
model.add(tf.layers.dense({units: 2, inputShape: [3], activation: 'relu6'}));
model.add(tf.layers.dense({units: 3, inputShape: [5], activation: 'sigmoid'}));

// Calling getLayer() method
const layer_0 = model.getLayer(NaN, 0);
const layer_1 = model.getLayer('denselayer', 1);
const layer_2 = model.getLayer(undefined, 2);

// Printing number of numbers in the weights
// of the layer_0, layer_1, and layer_2 
// using countParams() method
console.log(layer_0.countParams());
console.log(layer_1.countParams());
console.log(layer_2.countParams());