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

Tensorflow.js是Google脑部团队开发的开放源代码库，用于在浏览器或节点环境中运行深度学习神经网络。

.dispose()函数用于将张量放置到内存中。

用法:

tensor.dispose()

参数：此方法没有任何参数

返回值：空白

范例1：在此示例中，我们将使用tf.dispose函数来处理张量。

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
var tr = tf.tensor([1, 2, 3, 4, 5, 6, 7]);
  
// Disposing the tensor
tr.dispose()
  
// Trying to print it now
tr.print()

输出：

An error occured 
  Tensor is disposed

范例2：在此示例中，我们将创建一个等级2的张量，并尝试在使用tf.dispose函数之前和之后将其打印到。

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Create a new tensor
var tr2d = tf.tensor2d([1, 2, 3, 4], [2, 2]);
  
// Print the tensor
tr2d.print()
  
// Dispose the tensor
tr2d.dispose()
  
// Try to print it again
tr2d.print()

输出：

"Tensor
    [[1, 2],
     [3, 4]]"

范例3：在此示例中，我们将创建具有值，形状和数据类型的张量。然后，我们将使用tf.dispose()函数对其进行处理：

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
   
// Creting a value variable which
// stores the value
var value = ['1', '2', '3', '4', '5', '6']
   
// Creting a shape variable
// which stores the shape
var shape = [2, 3]
   
// Creating a d_Type variable
// which strores the data-type
var d_Type = 'string'
   
// Creating the tensor
var tr3d = tf.tensor(value, shape, d_Type)
   
// Printing the tensor
tr3d.print()

输出：

"Tensor
    [['1', '2', '3'],
     ['4', '5', '6']]"

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

相关用法

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