Tensorflow.js tf.layers apply() 方法

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

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

句法：

apply (inputs, kwargs)

参数：

• 输入：此参数将数组保存为输入。
• kwargs[Optional]：它是一个可选参数，包含要传递给 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