当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。