Open In App

Tensorflow.js tf.TensorBuffer Class .toTensor() Method

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.TensorBuffer class .toTensor() function is used to create an immutable Tensor object from the specified buffer and its values.



Syntax:

toTensor ()

Parameters: This function does not accept any parameters.



Return Value: It returns the created immutable Tensor object.

Example 1:




// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 2*2 dimensions
const buffer = tf.buffer([2, 2]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 1, 0); 
  
// Converting the above buffer back
// to a tensor value to print with 
// the help of .toTensor() function
buffer.toTensor().print();

Output:

Tensor
  [[5 , 0],
   [10, 0]]

Example 2:




// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 3*3 dimensions
const buffer = tf.buffer([3, 3]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 0, 1); 
buffer.set(15, 1, 0); 
buffer.set(20, 1, 1); 
buffer.set(25, 2, 0); 
buffer.set(30, 2, 1); 
buffer.set(35, 2, 2); 
  
// Converting the above buffer back 
// to a tensor value to print with
// the help of .toTensor() function
buffer.toTensor().print()

Output:

Tensor
  [[5 , 10, 0 ],
   [15, 20, 0 ],
   [25, 30, 35]]

Reference: https://js.tensorflow.org/api/latest/#tf.TensorBuffer.toTensor


Article Tags :