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

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

tf.layers apply() 方法用于执行层计算并在我们使用 tf.Tensor(s) 调用它时返回 Tensor(s)。如果我们使用 tf.SymbolicTensor(s) 调用它，那么将为未来执行准备层。

用法：

apply (inputs, kwargs)

参数：

inputs:此参数将数组作为输入。
kwargs[可选]：它是一个可选参数，包含要传递给 call() 的附加关键字参数。

返回值：它返回相同数据类型的张量或符号张量。

下面的例子说明了 Tensorflow.js tf.layers apply() 方法：

范例1：

Javascript


import * as tf from "@tensorflow/tfjs"
  
const denseLayer = tf.layers.dense({
   units:1,
   kernelInitializer:'ones',
   useBias:false
});
  
const input = tf.ones([2, 2]);
const output = denseLayer.apply(input);
  
// Print the output
print(output)

输出：

Tensor [[2], [2]]

范例2：

Javascript


import * as tf from "@tensorflow/tfjs"
  
const denseLayer = tf.layers.dense({
   units:1,
   kernelInitializer:'zeros',
   useBias:false
});
  
const input = tf.ones([2, 2]);
const output = denseLayer.apply(input);
  
// Print the output
print(output)

输出：

Tensor [[0], [0]]

参考：https://js.tensorflow.org/api/latest/#tf.layers.Layer.apply

相关用法

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