Tensorflow.js tf.util.sizeFromShape()函数

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

.util.sizeFromShape()函数用于查找元素的数量，即从所述形状输入的所述张量的大小。

句法 ：

tf.util.sizeFromShape(shape)

参数：

• 形状：它是要返回其大小的规定形状。它的类型为 number[]。

返回值：它返回一个数字。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Defining shape
const sh = [5, 8, 1];
 
// Calling tf.util.sizeFromShape() method
const num = tf.util.sizeFromShape(sh);
 
// printing output
console.log(num);

输出：

40

示例 2：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Calling tf.util.sizeFromShape() method and
// printing output
console.log(tf.util.sizeFromShape([1.4, 5.6, 9.0]));

输出：

70.55999999999999

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