當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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

Tensorflow.js是Google開發的一個開放源代碼庫,用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。。它還可以幫助開發人員使用JavaScript語言開發ML模型,並且可以直接在瀏覽器或Node.js中使用ML。

tf.split()函數用於將tf.tensor拆分為次張量。

用法:

tf.split (x, numOrSizeSplits, axis?)

參數:

  • x:要分割的輸入張量。
  • numOrSizeSplits:它可以是表示分割數的數字,也可以是為每個輸出張量提供大小的數組
  • axis:它是沿其分割的尺寸軸。

返回值:它返回tf.Tensor []。



範例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor2d([10, 20, 50, 60,
                       30, 40, 70, 80], [2, 4]);
  
// Split the tensor
const [gfg, geeks] = tf.split(x, 2, 1);
  
gfg.print();
  
geeks.print();

輸出:

Tensor
    [[10, 20],
     [30, 40]]
Tensor
    [[50, 60],
     [70, 80]]

範例2:在此示例中,將以軸作為第二個參數的分割作為數組。

Javascript


const x = tf.tensor2d([10, 30, 50, 70, 20, 40, 60, 80], [2, 4]);
  
const [gfg, gfg1, geeks] = tf.split(x, [1, 2, 1], 1);
  
gfg.print();
  
gfg1.print();
  
geeks.print();

輸出:

Tensor
    [[10],
     [20]]
Tensor
    [[30, 50],
     [40, 60]]
Tensor
    [[70],
     [80]]

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

相關用法


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