TensorFlow.js tf.min()函数介绍

简介

tf.min()函数是 TensorFlow.js 中的一个重要函数，用于计算张量中的最小值。它能够在 JavaScript 中进行高性能的数值计算，并且可以在浏览器和 Node.js 环境中运行。

语法

该函数的语法如下所示：

tf.min(x, axis)

• x：输入的张量。
• axis：可选参数，指定进行最小值计算的轴。如果提供了该参数，则返回指定轴上的最小值。

返回值

tf.min()函数返回一个新的张量，包含输入张量中的最小值。

示例代码

下面是一个使用tf.min()函数的示例代码：

// 在浏览器控制台中运行该代码，需要先引入 TensorFlow.js

// 创建一个张量
const x = tf.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]]);

// 计算张量中的最小值
const min = tf.min(x);

// 打印结果
min.print();

运行以上代码，将会输出以下结果：

1

指定轴的示例

如果想要指定进行最小值计算的轴，可以传入axis参数。下面是一个指定轴的示例代码：

// 在浏览器控制台中运行该代码，需要先引入 TensorFlow.js

// 创建一个张量
const x = tf.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]]);

// 沿第一个轴计算最小值
const minAlongAxis0 = tf.min(x, 0);

// 打印结果
minAlongAxis0.print();

// 沿第二个轴计算最小值
const minAlongAxis1 = tf.min(x, 1);

// 打印结果
minAlongAxis1.print();

运行以上代码，将会输出以下结果：

Tensor
  [[1, 2, 3]] // 沿第一个轴的最小值
  shape: [1, 3]

Tensor
  [[1], // 第一个轴上的最小值
   [4], // 第二个轴上的最小值
   [7]] // 第三个轴上的最小值
  shape: [3, 1]

总结

通过 tf.min() 函数，你可以方便地计算输入张量中的最小值，还可以根据需要指定计算的轴。这个函数在 TensorFlow.js 中非常实用，为 JavaScript 开发人员提供了强大的数值计算能力。