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

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

tf.concat()函數用於沿給定軸連接指定張量的列表。

用法:

tf.concat (tensors, axis)

參數：此函數接受兩個參數，如下所示：

tensors:它是要連接的指定張量的列表。
axis:它是執行串聯過程的軸。它是一個可選參數，默認值為0。

返回值：它返回串聯張量的張量。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing two tensors to cancatenate
const A = tf.tensor1d([0, 2, 4]);
const B = tf.tensor1d([1, 3, 5]);
  
// Calling the .concat() function over
// the above tensors as its parameters
A.concat(B).print();

輸出：

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

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing three 2-D tensors to cancatenate
const A = tf.tensor2d([[0, 2], [1, 3]]);
const B = tf.tensor2d([[4, 6], [5, 7]]);
const C = tf.tensor2d([[8, 10], [9, 11]]);
  
// Initializing axis parameter
const axis = 1;
  
// Calling the .concat() function over
// the above tensors and axis as its parameters
tf.concat([A, B, C], axis).print();

輸出：

Tensor
   [[0, 2, 4, 6, 8, 10],
    [1, 3, 5, 7, 9, 11]]

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

相關用法

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