TensorFlow.js tf.expandDims()函数

tf.expandDims()是TensorFlow.js中的一个函数，用于将张量的维度扩展。

描述

tf.expandDims()函数可以将张量的维度扩展1个轴。例如，将形状为[2, 3]的张量扩展为形状为[2, 1, 3]的张量。

tf.expandDims()函数的参数如下：

• input：需要进行维度扩展的张量。
• axis：需要扩展的轴。

例子

以下代码演示了如何使用tf.expandDims()函数将形状为[2, 3]的张量扩展为形状为[2, 1, 3]的张量：

const tensor = tf.tensor2d([[1, 2, 3], [4, 5, 6]]);
const expandedTensor = tf.expandDims(tensor, 1);
console.log(expandedTensor.shape);
// 输出: [2, 1, 3]

在上述代码中，tensor是需要进行维度扩展的张量，1表示需要在第二个轴进行维度扩展。

总结

tf.expandDims()函数用于将张量的维度扩展，当需要对特定轴进行扩展时，可以使用该函数实现。