Tensorflow.js tf.addN()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf.addN()函数用于按元素添加指定的Tensors列表。每个张量应具有相同的形状和数据类型。

用法:

tf.addN (tensors)

参数：此函数接受如下所示的参数：

tensors:具有相同形状和数据类型的指定张量的数组/列表。

返回值：它返回添加元素的张量。

范例1：

Javascript


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

输出：

Tensor
   [3, 5, 7]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using some Tensors as the parameter
// for the .addN() function 
tf.addN([[12, 5, 6], [3, 2, 5], [2, 7, 9]]).print();

输出：

Tensor
   [17, 14, 20]

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

相关用法

注：本文由纯净天空筛选整理自Kanchan_Ray大神的英文原创作品 Tensorflow.js tf.addN() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。