Tensorflow.js tf.cast() 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.
The tf.cast() function is used to cast a specified Tensor to a new data type.
Syntax:
tf.cast (x, dtype)
Parameters: This function accepts two parameters which are illustrated below:
- x: The input tensor which is being casted.
- dtype: The data type in which input tensor is going to be casted.
Return Value: It returns a casted tensor of new data type.
Example 1:
Javascript
import * as tf from "@tensorflow/tfjs"
const x = tf.tensor1d([2.3, 1.7, 5, 0, 1, 0.5]);
tf.cast(x, 'int32' ).print();
|
Output:
Tensor
[2, 1, 5, 0, 1, 0]
Example 2:
Javascript
import * as tf from "@tensorflow/tfjs"
tf.cast(tf.tensor1d([0, 1, -3]), 'bool' ).print();
|
Output:
Tensor
[false, true, true]
Reference: https://js.tensorflow.org/api/latest/#cast
Last Updated :
12 May, 2021
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...