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

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

tf.time() 函数用于执行所述函数 f() 并返回一个承诺，以便确定时间细节。结果是一个对象以及诸如 Wall 执行时间、内核执行时间之类的属性，不包括数据传输。如果使用了 WebGL 后端并且查询代码不可用，则此方法将抛出一个错误对象。此外，WebGL 后端将提供更多的属性，例如uploadWaitMs，即CPU 在纹理上传时的阻塞时间，以及downloadWaitMs，即CPU 在纹理下载时的阻塞时间。

用法:

tf.time(f)

参数：此函数接受单个参数，如下所述：

f:它是既要执行又要执行的既定函数。

返回值：它返回一个 TimingInfo 的 Promise。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using truncatedNormal() function in order to
// define the parameter for the function used
const p = tf.truncatedNormal([10, 10]);
  
// Calling time() method and also using
// dot() method
const tm = await tf.time(() => p.dot(p));
  
// Printing output
console.log(`Kernel execution time: 
  ${tm.kernelMs}, Wall execution time:${tm.wallMs}`
);

输出：

Kernel execution time:0.048498, Wall execution time:85.19999992847443

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling time() method and also using
// square() as well as ones() method
const t = await tf.time(() => tf.square(tf.ones([15, 21])));
  
// Printing output for WebGL backend
console.log(`uploadWaitMs:${t.uploadWaitMs},
  downloadWaitMs:${t.downloadWaitMs}`);

输出：

uploadWaitMs:1.100000023841858, downloadWaitMs:0

参考： https://js.tensorflow.org/api/latest/#time

相关用法

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