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

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

.computeOutputShape() 函数用于枚举所述层的输出形状。并且假定将创建图层以匹配提供的输入形状。

用法:

computeOutputShape(inputShape)

参数：

inputShape:它是规定的形状，即整数集或形状集列表。此外，Shape 元组可以包含 void 以支持自由大小，而不是整数。它可以是 ((null | number)[]|(null | number)[][]) 类型。

返回值：它返回 (null | number)[]|(null | number)[][]。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential();
  
// Adding a layer
model.add(tf.layers.dense({units:1, inputShape:[3]}));
  
// Defining inputShape
const inputShape = [6, 2, 6];
  
// Calling computeOutputShape() method with its 
// parameter
const val = model.layers[0].computeOutputShape(inputShape);
  
// Printing output
console.log(val);

输出：

6,2,1

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units:1, inputShape:[3]}));
model.add(tf.layers.dense({units:5}));
  
// Defining inputShape
const inputShape1 = [6, 2, 6, null];
const inputShape2 = [6.5, 2.6, 9.1, NaN];
  
  
// Calling computeOutputShape() method with its 
// parameter
const val1 = model.layers[0].computeOutputShape(inputShape1);
const val2 = model.layers[1].computeOutputShape(inputShape2);
  
// Printing output
console.log(val1);
console.log(val2);

输出：

6,2,6,1
6.5,2.6,9.1,5

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

相关用法

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