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

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

Tensorflow.js tf.layers.gru() 函数用于创建仅由一个 GRUCell 组成的 RNN 层，该层的 apply 方法对输入张量序列进行操作。输入张量的形状必须至少是 2D 的，并且第一维必须是时间步长。 gru 是门控循环单元。

用法:

tf.layers.gru(args)

参数：

args:它指定给定的配置对象。
1. recurrentActivation:它指定将用于循环步骤的激活函数。这个参数的默认值是hard sigmoid。
2. implementation:它规定了实施模式。它可以是 1 或 2。为了获得卓越的性能，建议实施。

返回值：它返回一个 tf.layers.Layer

范例1：

Javascript


// Importing the tensorflow.js library 
const tf = require("@tensorflow/tfjs"); 
  
// Create a RNN model with gru Layer
const RNN = tf.layers.gru({units:8, returnSequences:true});
  
// Create an input which will have 5 time steps
const input = tf.input({shape:[5, 10]});
const output = RNN.apply(input);
  
console.log(JSON.stringify(output.shape));

输出：

[null, 5, 8]

范例2：

Javascript


// Importing the tensorflow.js library 
const tf = require("@tensorflow/tfjs"); 
  
// Create a new model with gru Layer
const rnn = tf.layers.gru({units:4, returnSequences:true});
  
// Create a 3d tensor
const x = tf.tensor3d([
    [
        [1, 2],
        [3, 4],
    ],
    [
        [5, 6],
        [7, 8],
    ],
]);
  
// Apply gru layer to x
const output = rnn.apply(x);
  
// Print output
output.print()

输出：

参考：https://js.tensorflow.org/api/1.0.0/#layers.gru

相关用法

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