Tensorflow.js tf.layers.reshape() Function

Tensorflow.js tf.layers.repeatVector() Function

Last Updated : 31 May, 2021

Tensorflow.js is an open-source library developed by Google for running machine learning models and deep learning neural networks in the browser or node environment.

The tf.layers.repeatVector() function is used to repeat the input n number of times in a new specified dimension. It is an inbuilt function of TensorFlow's.js library.

Syntax:

tf.layers.repeatVector(n)

Parameters:

n: Integer, specifying the number of times the input will be repeated.

Return value: It returns the tf.layers.Layer

Example 1:

JavaScript

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

// Create a new model
const model = tf.sequential();
  
// Add repeatVector layer to the model
model.add(tf.layers.repeatVector({
    n: 5, inputShape: [2]}
));
  
const x = tf.tensor2d([[10, 15]]);
 
console.log(model.predict(x).shape)

Output:

1, 5, 2

Example 2:

JavaScript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Create a new model
const model = tf.sequential();
  
// Add repeatVector layer to the model
model.add(tf.layers.repeatVector(
      {n: 8, inputShape: [2]}
));
 
const x = tf.tensor2d([[0,1]]);
 
model.predict(x).print();

Output:

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

Reference: https://js.tensorflow.org/api/latest/#layers.repeatVector

Tensorflow.js tf.layers.reshape() Function

sahilkumar101

Improve

Article Tags :

Similar Reads

Tensorflow.js tf.layers.reshape() Function

Tensorflow.js is an open-source library that is being developed by Google for running machine learning models as well as deep learning neural networks in the browser or node environment. The tf.layers.reshape() function is used to Reshape an input to a certain shape. Syntax: tf.layers.reshape(args)

Tensorflow.js tf.layers.permute() Function

The layers in models are the basic blocks in constructions of models, for every layer perform some computation on input and output to the next layer.Â The tf.layers.permute() function is inherited from layers class and is used to permute the dimensions of the input in a given pattern, Also used to co

Tensorflow.js tf.layers.reLU() Function

Tensorflow.js is a Google-developed open-source toolkit for executing machine learning models and deep learning neural networks in the browser or on the node platform. It also enables developers to create machine learning models in JavaScript and utilize them directly in the browser or with Node.js.

Tensorflow.js tf.layers.separableConv2d() Function

Tensorflow.js is an open-source library developed by Google for running machine learning models and deep learning neural networks in the browser or node environment. It also helps the developers to develop ML models in JavaScript language and can use ML directly in the browser or in Node.js. The tf.

Tensorflow.js tf.layers.prelu() Function

Tensorflow.js is a Google-developed open-source toolkit for executing machine learning models and deep learning neural networks in the browser or on the node platform. It also enables developers to create machine learning models in JavaScript and utilize them directly in the browser or with Node.js.

Tensorflow.js tf.layers.rnn() Function

Tensorflow.js is an open-source library developed by Google for running machine learning models and deep learning neural networks in the browser or node environment.Â Tensorflow.js tf.layers.rnn() function is basically base class for recurrent layers.Â Syntax: tf.layers.rnn( args ); Parameters:Â arg

Tensorflow.js tf.layers.simpleRNN() Function

Tensorflow.js is a Google-developed open-source toolkit for executing machine learning models and deep learning neural networks in the browser or on the node platform. It also enables developers to create machine learning models in JavaScript and utilize them directly in the browser or with Node.js.

Tensorflow.js tf.layers.leakyReLU() Function

Tensorflow.js is a Google-developed open-source toolkit for executing machine learning models and deep learning neural networks in the browser or on the node platform. It also enables developers to create machine learning models in JavaScript and utilize them directly in the browser or with Node.js.

Tensorflow.js tf.layers.lstm() Function

Tensorflow.js is an open-source library that is being developed by Google for running machine learning models as well as deep learning neural networks in the browser or node environment. tf.layers.lstm() function is used for creating an RNN layer consisting of one LSTMCell and the apply method of LS

Tensorflow.js tf.layers.lstmCell() Function

Tensorflow.js is an open-source library that is being developed by Google for running machine learning models as well as deep learning neural networks in the browser or node environment. The layers.lstmCell() function is used for Cell class for LSTM. It is separate from the RNN subclass. Syntax: tf.