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

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

.dispose() 函数用于从内存中处理指定的 tf.Tensor。

用法:

dispose()

参数：此方法不包含任何参数。

返回值：它返回void。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a tensor
const t = tf.tensor([6, 7]);
  
// Calling dispose() method
t.dispose();
  
// Printing output
console.log("Tensor Disposed.")

输出：

Tensor Disposed.

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a 2d tensor
const tn = tf.tensor2d([6, 7], [1, 2]);
  
// Calling dispose() method
const y = tn.dispose();
  
// Printing output
console.log(tn)

输出：

An error occured on line:11
Tensor is disposed.

在这里，打印张量输出时发生错误，因为张量已经被处理掉了。

参考： https://js.tensorflow.org/api/latest/#tf.Tensor.dispose

相关用法

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