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


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


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

tf.layers.timeDistributed() 函数用于将层的包裹应用于指定输入的每个时间切片。给定的输入必须至少为 3D,维度的索引 1 将被视为时间维度。

用法:

tf.layers.timeDistributed(args)

参数:此函数接受单个参数 args,可用于指定以下属性:

  • layer:它是要包装的指定层。
  • inputShape:它用于创建一个输入层以在该层之前插入。当 inputShape 和 batchInputShape 都被定义时,batchInputShape 将被使用。此参数仅与输入层相关,即模型的第一层。它是一个可选参数。
  • batchInputShape:定义 inputShape 和 batchInputShape 时将使用此参数。此参数仅与输入层相关,即模型的第一层。它是一个可选参数。
  • batchSize:它用于在给定 inputShape 时创建 batchInputShape,但没有给定 batchInputShape。
  • dtype:它是该层的数据类型,其默认值为 ‘float32’。此参数仅与输入层相关,即模型的第一层。
  • name:这是该层的名称。
  • trainable:它的默认值为真。它表示该层的权重是否可以通过拟合更新。
  • weights:它是图层的第一个权重值。
  • inputDType:建议不要用于新代码。它用于遗留支持。

返回值:它返回一个 TimeDistributed 对象。



范例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a model
const model = tf.sequential();
  
// Calling the tf.layers.timeDistributed() function
const a = model.add(tf.layers.timeDistributed({
   layer:tf.layers.dense({units:8}),
    
   // Considering a sequence of 5 vectors
   // of 10 dimensions
   inputShape:[5, 10],
}));
  
// Getting the model.outputShape
console.log(JSON.stringify(model.outputs[0].shape));

输出:

[null,5,8]

范例2:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a model
const model = tf.sequential();
  
// Calling the tf.layers.timeDistributed() function
const a = model.add(tf.layers.timeDistributed({
    
   // Initializing the first layer with inputShape
   layer:tf.layers.dense({units:12}),
    
   // Considering a sequence of 5 vectors
   // of 10 dimensions
   inputShape:[5, 10],
}));
  
// In the second layer, there is no 
// need for `inputShape`
model.add(tf.layers.timeDistributed(
  {layer:tf.layers.dense({units:32})}
));
  
// Getting the model.outputShape
console.log(JSON.stringify(model.outputs[0].shape));

输出:

[null,5,32]

参考:https://js.tensorflow.org/api/latest/#layers.timeDistributed




相关用法


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