Tensorflow.js tf.argMin() Function

0 0 1 minute read

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.argMin() function is used to return the indices for the minimum values of the specified Tensor along an axis. The output result has the same shape as input with the dimension along axis removed.

Syntax:

tf.argMin (x, axis)

Parameters: This function accepts two parameters which are illustrated below:

x: The input tensor.
axis: The specified dimension(s) to reduce. It is an optional parameter and its default value is 0.

Return Value: It returns a Tensor of the indices of the minimum values along an axis.

Example 1:

Javascript

// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Initializing a some tensors  
const a = tf.tensor1d([0, 1]); 
const b = tf.tensor1d([5, 3]); 
const c = tf.tensor1d([6, 3, 5, 2]); 
  
// Calling the .argMin() function over  
// the above tensors 
a.argMin().print(); 
b.argMin().print(); 
c.argMin().print();

Output:

Tensor
   0
Tensor
   1
Tensor
   3

Example 2:

Javascript

// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Initializing a some tensors  
const a = tf.tensor1d([0, 1]); 
const b = tf.tensor2d([9, 5, 2, 8], [2, 2]); 
const c = tf.tensor1d([6, 4, 7]); 
  
// Initializing a axis parameters 
const axis1 = -1; 
const axis2 = -2; 
const axis3 = 0; 
  
// Calling the .argMin() function over  
// the above tensors 
a.argMin(axis1).print(); 
b.argMin(axis2).print(); 
c.argMin(axis3).print();

Output:

Tensor
   0
Tensor
   [1, 0]
Tensor
   1

Reference: https://js.tensorflow.org/api/latest/#argMin

Whether you’re preparing for your first job interview or aiming to upskill in this ever-evolving tech landscape, zambiatek 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!