Tensorflow.js tf.TensorBuffer.toTensor()用法及代碼示例

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。

tf.TensorBuffer 類 .toTensor() 函數用於從指定的緩衝區及其值創建一個不可變的 Tensor 對象。

用法:

toTensor ()

參數：該函數不接受任何參數。

返回值：它返回創建的不可變 Tensor 對象。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 2*2 dimensions
const buffer = tf.buffer([2, 2]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 1, 0); 
  
// Converting the above buffer back
// to a tensor value to print with 
// the help of .toTensor() function
buffer.toTensor().print();

輸出：

Tensor
  [[5 , 0],
   [10, 0]]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 3*3 dimensions
const buffer = tf.buffer([3, 3]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 0, 1); 
buffer.set(15, 1, 0); 
buffer.set(20, 1, 1); 
buffer.set(25, 2, 0); 
buffer.set(30, 2, 1); 
buffer.set(35, 2, 2); 
  
// Converting the above buffer back 
// to a tensor value to print with
// the help of .toTensor() function
buffer.toTensor().print()

輸出：

Tensor
  [[5 , 10, 0 ],
   [15, 20, 0 ],
   [25, 30, 35]]

參考： https://js.tensorflow.org/api/latest/#tf.TensorBuffer.toTensor

相關用法

注：本文由純淨天空篩選整理自Kanchan_Ray大神的英文原創作品 Tensorflow.js tf.TensorBuffer Class .toTensor() Method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。