Tensorflow.js is an open-source library for running machine learning models and deep learning neural networks in the browser or node environment. The tf.ones() function is used to create a new tensor where all elements are set to 1.
Syntax:
tf.ones(shape, dtype, name)
Parameters:
- shape: It takes the shape of the tensor. It can be a multi-dimensional array or an int32
- dtype: It takes the data type of the 1 we are inserting. It is by default set to float32, however, it can be int32 as well. It is an optional parameter.
- name: It takes the name of the operation we are performing. It is None by default. It is an optional parameter.
Return Value: It returns a tensor with the same data type.
Example 1: In this example, we will create a 2D Tensor with all values set to 1 of integer data type.
JavaScript
// Import tensorflow import tensorflow as tf   // Get a Tensor val = tf.ones([10, 10], tf.int32)   // Print the Tensor print(val) |
Output:
tf.Tensor( [[1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1] [1 1 1 1 1 1 1 1 1 1]], shape=(10, 10), dtype=int32)
Example 2: In this example, we will create a 1D Tensor with all values set to 1 of float data type.
JavaScript
// Import tensorflow import tensorflow as tf   // Get a Tensor val = tf.ones(5, tf.float32)   // Print a Tensor print(val) |
Output:
tf.Tensor([1. 1. 1. 1. 1.], shape=(5,), dtype=float32)
Reference:https://js.tensorflow.org/api/latest/#ones
Whether you’re preparing for your first job interview or aiming to upskill in this ever-evolving tech landscape, neveropen Courses are your key to success. We provide top-quality content at affordable prices, all geared towards accelerating your growth in a time-bound manner. Join the millions we’ve already empowered, and we’re here to do the same for you. Don’t miss out – check it out now!
