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

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

.tensor3d()函數用於創建具有參數(即值，形狀和數據類型)的新3維張量。

用法:

tf.tensor3d (value, shape, datatype)

Parameters:

value:張量的值，它可以是數字的嵌套數組，平麵數組或TypedArray。
shape:它采用張量的形狀。如果未提供張量，則張量將從其值推斷其形狀。它是一個可選參數。
datatype:它可以是‘float32’或‘int32’或‘bool’或‘complex64’或‘string’值。它是一個可選參數。

返回值：它返回相同數據類型的張量。返回的張量將始終是3維的。

注意：也可以使用tf.tensor()函數來實現3d張量函數，但是使用tf.tensor3d()可以使代碼易於理解和讀取。

範例1：

在這裏，我們正在創建3d張量並進行打印。為了創建3d張量，我們使用.tensor3d()函數，並使用.print()函數打印張量。在這裏，我們會將3d數組(即嵌套數組)傳遞給value參數。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs";
  
// Create the tensor
let example1 = tf.tensor3d([
    [
        [1, 2],
        [3, 4],
    ],
    [
        [5, 6],
        [7, 8],
    ],
]);
  
// Print the tensor
example1.print()

輸出：

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

    [[5, 6],  
     [7, 8]]]

範例2：

在此示例中，我們在此處創建張量，並在其中傳遞平麵數組並指定張量的shape參數。我們將在這裏看到shape參數的用法。

Javascript


// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs" 
  
// Define the value of the tensor
const value = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12];
  
// Specify the shape of the tensor
const shape = [2, 3, 2];
  
// Create the tensor
let example2 = tf.tensor3d(value, shape);
  
// Print the tensor
example2.print();

輸出：

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

     [[7 , 8 ],
      [9 , 10],
      [11, 12]]]

在上麵的示例中，我們創建了一個尺寸為2 x 3 x 2的張量。

範例3：

在這裏，在此示例中，我們將通過指定值，形狀和數據類型來創建張量。我們將創建張量，其中所有值均為字符串數據類型。

Javascript


// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs";
  
// Define the value of the tensor
const value = ["C", "C++", "Java", "Python", 
               "PHP", "JS", "SQL", "React"];
  
// Specify the shape of the tensor
const shape = [2, 2, 2];
  
// Create the tensor
let example3 = tf.tensor3d(value, shape);
  
// Print the tensor
example3.print();

輸出：

Tensor
    [[['C'   , 'C++'   ], 
      ['Java', 'Python']],

     [['PHP' , 'JS'    ], 
      ['SQL' , 'React' ]]]

在上麵的示例中，張量的打印值是字符串數據類型。

參考：https://js.tensorflow.org/api/latest/#tensor3d

相關用法

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