Tensorflow.js tf.mean()函数

简介

tf.mean()函数是Tensorflow.js库中的一种计算函数，它的作用是计算张量（Tensor）的平均值。这个函数可以接受多种类型的输出张量，并且可以指定维度来计算平均值。它可以在客户端JavaScript中使用，也可以用于服务器端JavaScript，如Node.js。

语法

tf.mean(x, axis?)

• x：需要计算平均值的张量。
• axis：可选参数，表示对应的轴（维度）来计算平均值。

例子

对一个一维数组取平均值

const values = [1, 2, 3, 4, 5];
const tensor = tf.tensor1d(values);
const mean = tensor.mean();
mean.print();

上述代码创建了一个长度为5的一维数组，并将其转换为一个一维张量。然后调用tf.mean()函数来计算张量中所有值的平均值。最后调用print()函数将其打印出来。

输出结果：

3

对一个二维数组取平均值

const values = [[1, 2], [3, 4], [5, 6]];
const tensor = tf.tensor2d(values);
const mean = tensor.mean(0);
mean.print();

上述代码创建了一个3行2列的二维数组，并将其转换为一个二维张量。然后调用tf.mean()函数指定按照第一个轴（即行）计算平均值。最后将结果打印出来。

输出结果：

Tensor
  [[3, 4]]

对一个三维数组取平均值

const values = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]];
const tensor = tf.tensor3d(values);
const mean = tensor.mean([1, 2]);
mean.print();

上述代码创建了一个2个2行2列的三维数组，并将其转换为一个三维张量。然后调用tf.mean()函数指定按照第2个和第3个轴（即列）计算平均值。最后将结果打印出来。

输出结果：

Tensor
  [ [ 2.5 ], [ 6.5 ] ]

总结

tf.mean()函数是Tensorflow.js库中一个很实用的计算函数。它可以用于各种类型的输出张量，并且可以指定维度来计算平均值。希望本文对您有所帮助。