Tensorflow.js tf.stack()函数

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它帮助开发人员使用 JavaScript 开发 ML 模型，并直接在浏览器或 Node.js 中使用 ML。

tf.stack()函数用于将 tf,tensor 堆栈创建为 r+1 秩 tf.tensor。

句法：

tf.stack(tensors, axis)

参数：此函数接受上面提到的两个参数，下面讨论。

• 张量：要使用的张量对象列表，具有相同的形状和数据类型。
• 轴：它是堆栈沿的轴。

返回值：返回 tf.Tensor。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Making a tensor a
const a = tf.tensor1d([99, 999, 999]);
  
// Making a tensor b
const b = tf.tensor1d([322, 411, 888]);
  
// Making a tensor c
const c = tf.tensor1d([523, 622, 666]);
  
// Printing the stack
tf.stack([a, b, c]).print();

输出：

Tensor
    [[99 , 999, 999],
     [322, 411, 888],
     [523, 622, 666]]

示例 2：在此示例中，使用轴作为第二个参数的堆栈。

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Making a tensor a
const a = tf.tensor1d([99, 999, 999]);
  
// Making a tensor b
const b = tf.tensor1d([322, 411, 888]);
  
// Making a tensor c
const c = tf.tensor1d([523, 622, 666]);
  
// Printing the stack
tf.stack([a, b, c], 1).print();

输出：

Tensor
    [[99 , 322, 523],
     [999, 411, 622],
     [999, 888, 666]]

参考： https://js.tensorflow.org/api/latest/#stack