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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。