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

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

.tensor6d()函數用於創建一個新的6維張量，其參數為值，形狀和數據類型。

句法：

tf.tensor6d(value, shape?, dataType?)

參數：

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

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

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

範例1：在這裏，我們將創建一個6d張量並進行打印。為了創建6d張量，我們使用.tensor6d()函數，並使用.print()函數打印張量。在這裏，我們會將6d數組(即嵌套數組)傳遞給value參數。

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

輸出：

Tensor
    [[[[[[1, 3],
         [2, 8]],

        [[3, 9],
         [4, 2]]]]]]

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

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

輸出：

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

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

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

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

輸出：

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

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

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