Tensorflow.js tf.TensorBuffer Class .toTensor() Method
Last Updated :
22 Apr, 2022
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
import * as tf from "@tensorflow/tfjs"
const buffer = tf.buffer([2, 2]);
buffer.set(5, 0, 0);
buffer.set(10, 1, 0);
buffer.toTensor().print();
|
Output:
Tensor
[[5 , 0],
[10, 0]]
Example 2:
Javascript
import * as tf from "@tensorflow/tfjs"
const buffer = tf.buffer([3, 3]);
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);
buffer.toTensor().print()
|
Output:
Tensor
[[5 , 10, 0 ],
[15, 20, 0 ],
[25, 30, 35]]
Reference: https://js.tensorflow.org/api/latest/#tf.TensorBuffer.toTensor
Share your thoughts in the comments
Please Login to comment...