Open In App
Related Articles

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

Improve Article
Improve
Save Article
Save
Like Article
Like

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:

Javascript




// 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:

Javascript




// 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


Last Updated : 22 Apr, 2022
Like Article
Save Article
Similar Reads
Related Tutorials